|
Mila 0.13.48
Deep Neural Network Library
|
Classes | |
| struct | Mila::Dnn::TensorHostTypeMap< TDataType > |
| Maps abstract TensorDataType to host-compatible C++ type and TensorDataType. More... | |
Typedefs | |
| template<TensorDataType TDataType> | |
| using | Mila::Dnn::host_type_t = typename TensorHostTypeMap<TDataType>::host_type |
| Convenience alias for accessing host type mapping. | |
Variables | |
| template<TensorDataType TDataType> | |
| constexpr bool | Mila::Dnn::is_host_float_type = std::is_floating_point_v<host_type_t<TDataType>> |
| Checks if a TensorDataType maps to a floating-point host type. | |
| template<TensorDataType TDataType> | |
| constexpr bool | Mila::Dnn::is_host_integer_type = std::is_integral_v<host_type_t<TDataType>> |
| Checks if a TensorDataType maps to an integer host type. | |
Files | |
| file | /__w/Mila/Mila/Mila/Src/Dnn/Tensors/TensorHostTypeMap.ixx |
| Device-agnostic host type mapping for abstract TensorDataType enumeration. | |