batchnorm_node.h File Reference

#include "sdfg/data_flow/library_nodes/math/tensor/tensor_layout.h"
#include "sdfg/data_flow/library_nodes/math/tensor/tensor_node.h"
#include "sdfg/serializer/json_serializer.h"

Go to the source code of this file.

Classes
class	sdfg::math::tensor::BatchNormNode
	In N, C, D0...Dn tensor, applies normalization to per channel in C across all the inner dimensions. More...
class	sdfg::math::tensor::BatchNormNodeSerializer

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

Functions
data_flow::LibraryNodeCode	sdfg::math::tensor::LibraryNodeType_BatchNorm ("ml::BatchNorm")