Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Compute::Cpu::MathOps Member List

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

add(const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b, Tensor< TDataType, TMemoryResource > &result, IExecutionContext *exec_context=nullptr)Mila::Dnn::Compute::Cpu::MathOpsinlinestatic
divide(const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b, Tensor< TDataType, TMemoryResource > &result, IExecutionContext *exec_context=nullptr)Mila::Dnn::Compute::Cpu::MathOpsinlinestatic
multiply(const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b, Tensor< TDataType, TMemoryResource > &result, IExecutionContext *exec_context=nullptr)Mila::Dnn::Compute::Cpu::MathOpsinlinestatic
performElementwiseOperation(const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b, Tensor< TDataType, TMemoryResource > &result, TBinaryOp op)Mila::Dnn::Compute::Cpu::MathOpsinlineprivatestatic
subtract(const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b, Tensor< TDataType, TMemoryResource > &result, IExecutionContext *exec_context=nullptr)Mila::Dnn::Compute::Cpu::MathOpsinlinestatic
sum(const Tensor< TDataType, TMemoryResource > &tensor, IExecutionContext *exec_context=nullptr)Mila::Dnn::Compute::Cpu::MathOpsinlinestatic
validateShapeCompatibility(const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b, const Tensor< TDataType, TMemoryResource > &result, const std::string &operation_name)Mila::Dnn::Compute::Cpu::MathOpsinlineprivatestatic