Norms.hpp File Reference

Normalisation operations. More...

#include "poplar/DebugContext.hpp"
#include "poplar/Program.hpp"
#include "poplar/Tensor.hpp"

Go to the source code of this file.

Namespaces
namespace	popnn
	Functions used in neural networks.

Functions
std::uint64_t	popnn::getNormFwdFlops (std::size_t statisticsSize, std::size_t numActsElements, bool computeStats=true)
	Calculate the floating point operations required for the forward pass of a norm layer. More...
std::uint64_t	popnn::getNormBwdFlops (std::size_t statisticsSize, std::size_t numActsElements)
	Calculate the floating point operations required for computation of the gradient with respect to the activations for a norm layer. More...
std::uint64_t	popnn::getNormWuFlops (std::size_t paramsSize, std::size_t numActsElements)
	Calculate the floating point operations required for parameter update for a norm layer. More...
poplar::Tensor	popnn::createNormGamma (poplar::Graph &graph, const poplar::Tensor &acts, const poplar::DebugContext &debugContext={})
poplar::Tensor	popnn::createNormBeta (poplar::Graph &graph, const poplar::Tensor &acts, const poplar::DebugContext &debugContext={})
std::pair< poplar::Tensor, poplar::Tensor >	popnn::createNormParams (poplar::Graph &graph, const poplar::Tensor acts, const poplar::DebugContext &debugContext={})

Detailed Description

Normalisation operations.