|
Mila 0.13.48
Deep Neural Network Library
|
Abstract device-agnostic paired operation interface. More...
#include <type_traits>import Compute.CpuMemoryResource;import Compute.OperationBase;import Compute.DeviceTypeTraits;import Compute.DeviceType;import Compute.Device;import Compute.ExecutionContext;import Compute.OperationType;import Dnn.TensorDataTypeTraits;import Compute.IExecutionContext;import Dnn.TensorDataType;import Dnn.ITensor;import Dnn.Tensor;Classes | |
| class | Mila::Dnn::Compute::PairedOperation< TDeviceType, TPrecision, TInputA, TInputB > |
| Abstract base for paired operations: two inputs -> two outputs. More... | |
Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::Dnn |
| namespace | Mila::Dnn::Compute |
Abstract device-agnostic paired operation interface.
Typed, device-aware interface for paired operations (two inputs -> two outputs). Models symmetric transforms such as RoPE and its variants.