Compute.OperationRegistrarHelpers Module Reference

Functions
template<DeviceType TDataType, typename OpType, Dnn::TensorDataType TA, Dnn::TensorDataType TB, Dnn::TensorDataType TP = TA>
void	Mila::Dnn::Compute::registerBinaryOpType (const std::string &opName)
	Register a binary operation type with OperationRegistry using a common factory pattern.
template<DeviceType TDataType, typename OpType, Dnn::TensorDataType TA, Dnn::TensorDataType TB = TA, Dnn::TensorDataType TP = TA>
void	Mila::Dnn::Compute::registerPairedOpType (const std::string &opName)
	Register a paired operation type with OperationRegistry using a common factory pattern.
template<DeviceType TDataType, typename OpType, Dnn::TensorDataType TInput, Dnn::TensorDataType TPrecision = TInput>
void	Mila::Dnn::Compute::registerUnaryOpType (std::string_view op_name)
	Register a unary operation type with OperationRegistry using a common factory pattern.

Files
file	/__w/Mila/Mila/Mila/Src/Dnn/Compute/Operations/OperationRegistrarHelpers.ixx
	Helpers to standardize registration of unary/binary/paired ops.