Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Compute::Cuda::MultiHeadAttention Namespace Reference

Namespaces

namespace  Detail

Classes

class  CudaMultiHeadAttentionOp
 CUDA implementation of Multi-Head Attention using column-major cuBLASLt optimization. More...
class  CudaMultiHeadAttentionOpRegistrar