pooling_node.h File Reference

#include "sdfg/data_flow/library_nodes/math/tensor/tensor_node.h"
#include "sdfg/codegen/dispatchers/block_dispatcher.h"
#include "sdfg/data_flow/library_nodes/math/tensor/spatial_tensor_node.h"
#include "sdfg/serializer/json_serializer.h"

Go to the source code of this file.

Classes
class	sdfg::math::tensor::PoolingNode
	N-dimensional pooling operation. More...
class	sdfg::math::tensor::PoolingNodeSerializer
	Serializer for PoolingNode. More...

Namespaces
namespace	sdfg
namespace	sdfg::math
namespace	sdfg::math::tensor

Enumerations
enum class	sdfg::math::tensor::PoolingMode { sdfg::math::tensor::Max , sdfg::math::tensor::Sum , sdfg::math::tensor::Avg }
	Pooling operation type. More...

Functions
data_flow::LibraryNodeCode	sdfg::math::tensor::LibraryNodeType_Pooling ("ml::Pooling")