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

Exported Modules

module  Compute.DeviceTypeTraits
module  Dnn.TensorTypes
module  Dnn.TensorDataType
module  Dnn.ITensor
module  Compute.DeviceType
module  Serialization.Mode
module  Serialization.Tensor
module  Compute.ExecutionContextFactory
module  Compute.OperationTraits
module  Compute.IExecutionContext
module  Serialization.ModelArchive
module  Dnn.TensorDataTypeTraits
module  Dnn.Components.ActivationConfig
module  Dnn.Component
module  Compute.CpuMemoryResource
module  Dnn.ComponentType
module  Dnn.ActivationType
module  Compute.DeviceId
module  Compute.Device
module  Dnn.Tensor

Classes

class  Mila::Dnn::Activation< TDeviceType, TPrecision, TFn >
 Unified elementwise activation component. More...
struct  Mila::Dnn::functor_of< TFn >
 Compile-time map from an elementwise ActivationType to its functor. More...
struct  Mila::Dnn::functor_of< ActivationType::Gelu >
struct  Mila::Dnn::functor_of< ActivationType::LeakyRelu >
struct  Mila::Dnn::functor_of< ActivationType::Mish >
struct  Mila::Dnn::functor_of< ActivationType::None >
struct  Mila::Dnn::functor_of< ActivationType::Relu >
struct  Mila::Dnn::functor_of< ActivationType::Sigmoid >
struct  Mila::Dnn::functor_of< ActivationType::Silu >
struct  Mila::Dnn::functor_of< ActivationType::Tanh >

Typedefs

template<ActivationType TFn>
using Mila::Dnn::functor_of_t = typename functor_of<TFn>::type

Files

file  Mila/Src/Dnn/Components/Activations/Activation/Activation.ixx
 Unified elementwise activation component.