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

Concepts

concept  Mila::Dnn::DeviceOnlyTensorDataType
 Concept identifying device-only abstract data types.
concept  Mila::Dnn::HostCompatibleTensorDataType
 Concept identifying host-compatible abstract data types.
concept  Mila::Dnn::isValidTensor
 Primary tensor configuration validation concept.
concept  Mila::Dnn::PrecisionSupportedOnDevice
 Concept to validate precision is supported on a device at compile-time.
concept  Mila::Dnn::ValidFloatTensorDataType
 Concept constraining abstract data types to floating-point formats.
concept  Mila::Dnn::ValidIntegerTensorDataType
 Concept constraining abstract data types to integer formats.

Classes

struct  Mila::Dnn::TensorDataTypeTraits< TDataType >
 Compile-time traits for TensorDataType enumeration values. More...

Typedefs

template<TensorDataType TDataType>
using Mila::Dnn::host_value_t = std::conditional_t<TensorDataTypeTraits<TDataType>::is_integer_type, int32_t, float>
 Host value type for given abstract tensor data type.

Files

file  /__w/Mila/Mila/Mila/Src/Dnn/Tensors/TensorDataTypeTraits.ixx
 Compile-time traits for the abstract TensorDataType enumeration.