Mila
Deep Neural Network Library
Loading...
Searching...
No Matches
Operations Directory Reference

Files

 
GqaState.ixx
 Non-owning transient scratch state for CudaGqaOp inference paths.
 
IKVCacheLifecycle.ixx
 Interface for operations that own and manage a KV cache.
 
IKvInference.ixx
 KV-cache compute interface for modern attention backends (GQA and beyond).
 
IPackedKvInference.ixx
 KV-cache inference interface for GPT-style packed-QKV MHA backends.
 
IPositionalDecode.ixx
 Interface for unary operations whose output depends on absolute token position.
 
IPositionalPairedOp.ixx
 Interface for paired operations whose output depends on absolute token position.
 
OperationBase.ixx
 Core abstraction for neural network operations in the Mila framework.
 
OperationTraits.ixx
 Aggregator for the unified operation traits dispatch table.
 
OperationTraits.Template.ixx
 Unified compile-time dispatch template mapping (OperationType, DeviceType, TPrecision, TPolicy) to a concrete operation type.
 
OperationType.ixx
 Defines the operation types supported by the compute framework.