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

Concepts

concept  Mila::Dnn::ValidFloatTensorType
 Concept that constrains types to valid floating-point tensor types.
 
concept  Mila::Dnn::ValidFloatTensorTypes
 Concept that verifies both types are valid floating-point tensor types.
 
concept  Mila::Dnn::ValidTensorType
 Concept that constrains types to those with valid tensor trait specializations.
 
concept  Mila::Dnn::ValidTensorTypes
 Concept that verifies both input and compute types have valid tensor trait mappings.
 

Classes

struct  Mila::Dnn::TensorTrait< T >
 Primary template for tensor type traits. More...
 

Functions

template<typename T >
constexpr std::string_view Mila::Dnn::tensor_type_name ()
 Get the string representation of a tensor element type.
 
template<typename T >
constexpr size_t Mila::Dnn::tensor_type_size ()
 Get the size in bytes of a tensor element type.
 

Files

file  /home/runner/work/Mila/Mila/Mila/Src/Dnn/Tensor/TensorTraits.ixx
 Provides type traits for tensor data types with compile-time type information.