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

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.