Mila Module Reference

Exported Modules
module	Dnn.Modules.Residual
module	Compute.CudaEncoderOp
module	Utils.Logger
module	Compute.OperationBase
module	Compute.CpuMemoryResource
module	Compute.CpuResidualOp
module	Compute.CpuGeluOp
module	Compute.CpuSoftmaxOp
module	Compute.CudaMHAOp
module	Dnn.Blocks.MLP
module	Core.RandomGenerator
module	Dnn.Modules.Softmax
module	Compute.CudaLayerNormOp
module	Dnn.ActivationType
module	Dnn.ComponentConfig
module	Dnn.Modules.Attention
module	Dnn.Tensor
module	Dnn.CompositeModule
module	Compute.DeviceRegistrar
module	Data.DataLoader
module	Compute.DeviceType
module	Compute.CudaResidualOp
module	Compute.CudaDevice
module	Compute.UnaryOperation
module	Cuda.Error
module	Compute.DeviceRegistry
module	Dnn.Module
module	Compute.BinaryOperation
module	Utils.DefaultLogger
module	Dnn.Modules.Linear
module	Dnn.Modules.Encoder
module	Compute.CpuDevice
module	Compute.DeviceHelpers
module	Compute.Precision
module	Compute.CpuLinearOp
module	Compute.OperationType
module	Compute.CpuLayerNormOp
module	Dnn.Gpt2.DatasetReader
module	Dnn.Modules.Gelu
module	Dnn.TensorBuffer
module	Dnn.TensorTraits
module	Dnn.Model
module	Compute.ComputeDevice
module	Dnn.Modules.LayerNorm
module	Compute.MemoryResource
module	Dnn.Blocks.TransformerBlock
module	Compute.OperationAttributes
module	Compute.CudaMemoryResource
module	Compute.DynamicMemoryResource
module	Dnn.TensorHelpers
module	Cuda.Helpers
module	Compute.DeviceContext
module	Compute.CudaManagedMemoryResource
module	Compute.CudaSoftmaxOp
module	Compute.CudaPinnedMemoryResource
module	Compute.OperationRegistry
module	Compute.CpuEncoderOp
module	Serialization.ModelArchive
module	Compute.CudaLinearOp
module	Compute.OperationsRegistrar
module	Compute.CudaGeluOp

Functions
Version	Mila::getAPIVersion ()
	Gets the current Mila API version.
bool	Mila::initialize (unsigned int randomSeed=0)
	Initializes the Mila framework.
void	Mila::initializeLogger (Utils::LogLevel level=Utils::LogLevel::Info)
void	Mila::shutdown ()

Variables
std::shared_ptr< Utils::DefaultLogger >	Mila::detail::g_defaultLogger

Files
file	/home/runner/work/Mila/Mila/Mila/Src/Mila.ixx