Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Dnn.TensorDataType Module Reference

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.