|
Mila 0.13.48
Deep Neural Network Library
|
Enumerations | |
| enum class | Mila::Dnn::TensorDataType { FP32 , FP16 , BF16 , FP8_E4M3 , FP8_E5M2 , FP4_E2M1 , FP4_E3M0 , INT8 , INT16 , INT32 , UINT8 , UINT16 , UINT32 } |
| Enumeration of supported abstract tensor data types. More... | |
Typedefs | |
| using | Mila::Dnn::dtype_t = TensorDataType |
| Alias for TensorDataType enumeration. | |
Functions | |
| TensorDataType | Mila::Dnn::parseTensorDataType (const std::string &type_str) |
| std::string | Mila::Dnn::tensorDataTypeToString (TensorDataType type) |
| Converts TensorDataType enumeration to human-readable string. | |
Files | |
| file | /__w/Mila/Mila/Mila/Src/Dnn/Tensors/TensorDataType.ixx |
| Abstract tensor data type enumeration and traits system for device-agnostic tensor operations. | |