Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Compute.OperationRegistryHelpers Module Reference

Functions

template<DeviceType TDevice, TensorDataType TDataType>
std::vector< std::string > Mila::Dnn::Compute::getRegisteredOperations ()
 Templated helper returning registered operation names for a compile-time device and tensor data type.
template<DeviceType TDevice, TensorDataType TDataType>
bool Mila::Dnn::Compute::isOperationRegistered (const std::string &operation_name)
 Templated helper that checks whether a named operation is registered for a compile-time device and tensor data type.

Files

file  /__w/Mila/Mila/Mila/Src/Dnn/Compute/Operations/OperationRegistryHelpers.ixx
 Compile-time templated helpers for querying the OperationRegistry.