Loading...
Searching...
No Matches
Here is a list of all functions with links to the classes they belong to:
- u -
- 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 >
- useBias() : Mila::Dnn::GptBlockConfig, Mila::Dnn::LlamaConfig
- usesPinnedMemory() : Mila::Data::DataLoader< TInputDataType, TTargetDataType, TMemoryResource >
- usesSameMaskPerBatch() : Mila::Dnn::DropoutConfig
- utf8CharLength() : Mila::Data::BpeTokenizer