Dnn.TensorPartitioning Module Reference

Classes
struct	Mila::Dnn::AxisPartition
	Information about axis partitioning of a tensor. More...
struct	Mila::Dnn::MultiAxisPartition
	Multi-axis partition for normalization over trailing dimensions. More...

Functions
AxisPartition	Mila::Dnn::computeAxisPartition (const shape_t &shape, dim_t axis, const char *op_name="Operation")
	Normalize and validate an axis, then compute partition sizes.
MultiAxisPartition	Mila::Dnn::computeNormalizedShapePartition (const shape_t &shape, const shape_t &normalized_shape, const char *op_name="Operation")
	Compute partition for normalization over trailing dimensions.
int64_t	Mila::Dnn::computeNumElements (const shape_t &shape)
	Compute total number of elements in a tensor shape.
void	Mila::Dnn::validateTensorSize (const shape_t &shape, int64_t expected_size, const char *tensor_name="tensor", const char *op_name="Operation")
	Validate that a tensor has the expected number of elements.

Files
file	/__w/Mila/Mila/Mila/Src/Dnn/Tensors/Tensor.Partitioning.ixx