|
Mila 0.13.48
Deep Neural Network Library
|
#include <memory>#include <vector>#include <string>#include <iostream>#include <sstream>#include <type_traits>#include <stdexcept>#include <cstdint>#include <optional>#include <cmath>#include <numeric>#include <algorithm>import Dnn.TensorHelpers;import Serialization.Tensor;import Compute.OperationTraits;import Compute.UnaryOperation;import Serialization.Mode;import Compute.ExecutionContextFactory;import Logging.Logger;import Dnn.Components.LpeConfig;import Compute.IPositionalDecode;import Compute.Device;import Compute.DeviceTypeTraits;import Dnn.Component;import Dnn.TensorDataType;import Dnn.ComponentType;import Dnn.TensorOps;import Compute.CpuMemoryResource;import Compute.MemoryResource;import Compute.DeviceType;import Dnn.Tensor;import Serialization.ModelArchive;import Dnn.TensorDataTypeTraits;import Dnn.ITensor;import Dnn.TensorTypes;import Compute.DeviceId;import Compute.ExecutionContext;Classes | |
| class | Mila::Dnn::Lpe< TDeviceType, TIndex, TPrecision > |
| Encoder module for token and positional embeddings (device-templated). More... | |
Namespaces | |
| namespace | Mila |
| Mila main API namespace. | |
| namespace | Mila::Dnn |