|
Mila 0.13.48
Deep Neural Network Library
|
Device-dispatched fast zero operation for tensor buffers. More...
#include <type_traits>import Compute.ExecutionContext;import Compute.DeviceType;import Dnn.TensorOps.Base;import Dnn.TensorDataTypeTraits;import Dnn.TensorDataTypeMap;import Dnn.TensorDataType;import Dnn.Tensor;Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::Dnn |
Functions | |
| template<TensorDataType TDataType, typename TMemoryResource> requires isValidTensor<TDataType, TMemoryResource> | |
| void | Mila::Dnn::zero (Tensor< TDataType, TMemoryResource > &tensor, IExecutionContext *exec_context=nullptr) |
| Zero a tensor using the fastest backend implementation. | |
Device-dispatched fast zero operation for tensor buffers.
Provides the high-level, device-agnostic entry point zero(...) that forwards to the backend TensorOps<device>::zero(...) implementation.