Pooling.hpp File Reference

Support for pooling operations. More...

#include <cstdint>
#include <poplar/Graph.hpp>
#include <poplar/OptionFlags.hpp>
#include <poplar/Program.hpp>
#include <popnn/PoolingDef.hpp>
#include <tuple>

Go to the source code of this file.

Namespaces
namespace	popnn
	Functions used in neural networks.
namespace	popnn::pooling
	Functions for pooling operations.

Functions
poplar::Tensor	popnn::pooling::pool (poplar::Graph &graph, const PoolParams &params, const poplar::Tensor &in, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={})
	Add a pooling operation to the graph. More...
poplar::Tensor	popnn::pooling::poolInputGradient (poplar::Graph &graph, const PoolParams &params, const poplar::Tensor &in, const poplar::Tensor &pooled, const poplar::Tensor &pooledGradient, bool useScaledGradient, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={})
	Calculate the gradient with respect to the input of a pooling operation given the gradient of the output. More...
poplar::Tensor	popnn::pooling::poolInputGradient (poplar::Graph &graph, const PoolParams &params, const unsigned fwdChansPerGroup, const poplar::Tensor &pooledGradient, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}, const poplar::OptionFlags &options={})
	Calculate the gradient with respect to the input of a pooling operation given the gradient of the output. More...

Detailed Description

Support for pooling operations.