Mila
Deep Neural Network Library
|
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. | |