Mila::Dnn::Compute::OperationRegistry Member List

This is the complete list of members for Mila::Dnn::Compute::OperationRegistry, including all inherited members.

createBinaryOperation(const std::string &operation_name, std::shared_ptr< DeviceContext > context, const ComponentConfig &config) const	Mila::Dnn::Compute::OperationRegistry	inline
createUnaryOperation(const std::string &operation_name, std::shared_ptr< DeviceContext > context, const ComponentConfig &config) const	Mila::Dnn::Compute::OperationRegistry	inline
findFusedMatch(const std::vector< std::string > &child_types, DeviceType device_type, std::type_index precision_type, const std::string &variant="Default")	Mila::Dnn::Compute::OperationRegistry	inline
fused_ops_	Mila::Dnn::Compute::OperationRegistry	private
GenericCreator typedef	Mila::Dnn::Compute::OperationRegistry	private
instance()	Mila::Dnn::Compute::OperationRegistry	inlinestatic
OperationRegistry()=default	Mila::Dnn::Compute::OperationRegistry	private
registerBinaryOperation(const std::string &operation_name, std::function< std::shared_ptr< BinaryOperation< TDeviceType, TInput1, TInput2, TOutput > >(std::shared_ptr< DeviceContext >, const ComponentConfig &)> creator)	Mila::Dnn::Compute::OperationRegistry	inline
registerFusedOperation(const std::vector< OperationType > &operation_types, const std::string &fused_op_name, const std::string &variant="Default")	Mila::Dnn::Compute::OperationRegistry	inline
registerUnaryOperation(const std::string &operation_name, std::function< std::shared_ptr< UnaryOperation< TDeviceType, TInput, TOutput > >(std::shared_ptr< DeviceContext >, const ComponentConfig &)> creator)	Mila::Dnn::Compute::OperationRegistry	inline
registry_	Mila::Dnn::Compute::OperationRegistry	private