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

Namespaces

namespace  Detail

Classes

class  CudaRopeOp
 CUDA implementation of the Rope (rotary positional embedding) operation. More...
class  CudaRopeOpRegistrar
class  RopeCacheRegistry
 Process-wide shared cache for RoPE cos/sin frequency tables. More...