Here is a list of all class members with links to the classes they belong to:
- u -
- uid_ : Mila::Dnn::Tensor< TDataType, TMemoryResource >
- unary_registry_ : Mila::Dnn::Compute::OperationRegistry
- UnaryOperationBase : Mila::Dnn::Compute::CpuGeluOp, Mila::Dnn::Compute::CpuLayerNormOp, Mila::Dnn::Compute::CpuLinearOp, Mila::Dnn::Compute::CpuSoftmaxOp, Mila::Dnn::Compute::Cuda::Gelu::CudaGeluOp< TPrecision >, Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >, Mila::Dnn::Compute::Cuda::LayerNorm::CudaLayerNormOp< TPrecision >, Mila::Dnn::Compute::Cuda::Lpe::CudaLpeOp< TInput, TPrecision >, Mila::Dnn::Compute::Cuda::MatMulBiasGelu::CudaMatMulBiasGeluOp< TInput, TOutput >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::CudaMultiHeadAttentionOp< TPrecision >, Mila::Dnn::Compute::Cuda::RmsNorm::CudaRmsNormOp< TPrecision >, Mila::Dnn::Compute::Cuda::Softmax::CudaSoftmaxOp< TPrecision >, Mila::Dnn::Compute::Cuda::Swiglu::CudaSwigluOp< TPrecision >, Mila::Dnn::Compute::Cuda::TokenEmbedding::CudaTokenEmbeddingOp< TInput, TPrecision >
- underlying_ : Mila::Dnn::Compute::TrackedMemoryResource
- unk_token : Mila::Data::SpecialTokens
- unk_token_id_ : Mila::Data::CharVocabulary
- unkTokenId() : Mila::Data::CharVocabulary
- unloadLibrary() : Mila::Dnn::Extensibility::PluginManager
- unpermute() : Mila::Dnn::Compute::CpuAttentionOp
- unpermute_backward() : Mila::Dnn::Compute::CpuAttentionOp, Mila::Dnn::Compute::Cuda::Gqa::Detail::cuda_gqa_kernels< float >, Mila::Dnn::Compute::Cuda::Gqa::Detail::cuda_gqa_kernels< nv_bfloat16 >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail::cuda_mha_kernels< float >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail::cuda_mha_kernels< half >
- unpermute_output() : Mila::Dnn::Compute::Cuda::Gqa::Detail::cuda_gqa_kernels< float >, Mila::Dnn::Compute::Cuda::Gqa::Detail::cuda_gqa_kernels< nv_bfloat16 >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail::cuda_mha_kernels< float >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail::cuda_mha_kernels< half >
- unpermute_output_padded() : Mila::Dnn::Compute::Cuda::Gqa::Detail::cuda_gqa_kernels< float >, Mila::Dnn::Compute::Cuda::Gqa::Detail::cuda_gqa_kernels< nv_bfloat16 >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail::cuda_mha_kernels< float >, Mila::Dnn::Compute::Cuda::MultiHeadAttention::Detail::cuda_mha_kernels< half >
- updateParameter() : Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >
- use_bias : Mila::Dnn::Serialization::PretrainedMetadata
- use_bias_ : Mila::Dnn::GptBlockConfig, Mila::Dnn::GptConfig, Mila::Dnn::LlamaConfig
- use_bos : Mila::Data::SpecialTokens
- use_cls : Mila::Data::SpecialTokens
- use_cublaslt_ : Mila::Dnn::Compute::Cuda::Linear::CudaLinearOp< TComputePrecision, TWeightQuant >
- use_eos : Mila::Data::SpecialTokens
- use_loop_unroll_ : Mila::Dnn::Compute::CpuLinearOp
- use_mask : Mila::Data::SpecialTokens
- use_optimized_kernel_ : Mila::Dnn::Compute::Cuda::Softmax::CudaSoftmaxOp< TPrecision >
- use_optimized_path_ : Mila::Dnn::Compute::Cuda::Gqa::CudaGqaOp< TPrecision >
- use_pad : Mila::Data::SpecialTokens
- use_same_mask_per_batch_ : Mila::Dnn::DropoutConfig
- use_sep : Mila::Data::SpecialTokens
- use_unk : Mila::Data::SpecialTokens
- use_wmma_fp4_gemm_ : Mila::Dnn::Compute::Cuda::Linear::CudaLinearOp< TComputePrecision, TWeightQuant >
- useBias() : Mila::Dnn::GptBlockConfig, Mila::Dnn::LlamaConfig
- uses_pinned_memory : Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
- usesPinnedMemory() : Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
- usesSameMaskPerBatch() : Mila::Dnn::DropoutConfig
- utf8CharLength() : Mila::Data::BpeTokenizer