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

This is the complete list of members for Mila::Dnn::Compute::Cuda::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::Cuda::MathOpsinlinestatic
addImpl(const void *a_data, const void *b_data, void *result_data, size_t count, cudaStream_t stream, int device_id)Mila::Dnn::Compute::Cuda::MathOpsinlineprivatestatic
divide(const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b, Tensor< TDataType, TMemoryResource > &result, IExecutionContext *exec_context=nullptr)Mila::Dnn::Compute::Cuda::MathOpsinlinestatic
divideImpl(const void *a_data, const void *b_data, void *result_data, size_t count, cudaStream_t stream, int device_id)Mila::Dnn::Compute::Cuda::MathOpsinlineprivatestatic
multiply(const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b, Tensor< TDataType, TMemoryResource > &result, IExecutionContext *exec_context=nullptr)Mila::Dnn::Compute::Cuda::MathOpsinlinestatic
multiplyImpl(const void *a_data, const void *b_data, void *result_data, size_t count, cudaStream_t stream, int device_id)Mila::Dnn::Compute::Cuda::MathOpsinlineprivatestatic
subtract(const Tensor< TDataType, TMemoryResource > &a, const Tensor< TDataType, TMemoryResource > &b, Tensor< TDataType, TMemoryResource > &result, IExecutionContext *exec_context=nullptr)Mila::Dnn::Compute::Cuda::MathOpsinlinestatic
subtractImpl(const void *a_data, const void *b_data, void *result_data, size_t count, cudaStream_t stream, int device_id)Mila::Dnn::Compute::Cuda::MathOpsinlineprivatestatic
sum(const Tensor< TDataType, TMemoryResource > &tensor, IExecutionContext *exec_context=nullptr)Mila::Dnn::Compute::Cuda::MathOpsinlinestatic
sumImpl(const void *tensor_data, size_t count, cudaStream_t stream, int device_id)Mila::Dnn::Compute::Cuda::MathOpsinlineprivatestatic