| backward(const TokenIndexType &input, const EmbeddingTensorType &output_grad) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inline |
| build(const BuildContext &context) final | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inlinevirtual |
| build_context_ | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | protected |
| Component(const std::string &name) | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inlineexplicit |
| ComponentBase typedef (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | |
| EmbeddingTensorType typedef (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | |
| forward(const TokenIndexType &input) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inline |
| getDeviceId() const override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| getDeviceType() | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inlinestatic |
| getEmbeddingDim() const noexcept (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inline |
| getExecutionContext() const | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inline |
| getGradients() const override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| getMemoryStats() const override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| getName() const | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inline |
| getObservableStages() const override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| getOutputs() const override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| getParameterNames() const override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| getParameters() const override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| getPrecision() noexcept | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inlinestatic |
| getRequiredMemory(const BuildContext &context) const override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| getTrainingMode() const noexcept | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inline |
| getType() const override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| getVocabSize() const noexcept (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inline |
| getWeightScalesTensorShared() const noexcept (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inline |
| getWeightTensorShared() const noexcept (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inline |
| getWteGrad() const noexcept (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inline |
| hasExecutionContext() const noexcept | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inlineprotected |
| HostStagingMemoryResource typedef | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | protected |
| isBuilt() const final | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inlinevirtual |
| kIsHostResident (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | static |
| kIsQuantized (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | static |
| kTableDtype (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | static |
| load_(ModelArchive &archive, SerializationMode mode) | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inlinevirtual |
| loadParameter(const std::string &name, const ITensorBlob &blob) override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| loadParameterFromBlob(const std::string ¶m_name, const Serialization::ITensorBlob &blob, Tensor< TParameterPrecision, TMemoryResource > &target, const shape_t &expected_shape) | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inlineprotected |
| MR typedef (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | |
| onBuilding(const BuildContext &build_context) override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlineprotectedvirtual |
| onExecutionContextSet() override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlineprotectedvirtual |
| onTrainingModeChanging(TrainingMode training_mode) override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlineprotectedvirtual |
| parameterCount() const override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| publish(Compute::ComputePass pass, std::string_view stage, const ITensor &value) const | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inlineprotected |
| requireSerializableParameters() const | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inlinevirtual |
| save_(ModelArchive &archive, SerializationMode mode) const override (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| saveFlatTensors(Serialization::SafeTensorsWriter &writer, const std::string &prefix, Serialization::TensorSavePass pass) const override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| saveParameterToArchive(ModelArchive &archive, const std::string ¶meter_name, const Tensor< TParameterPrecision, TMemoryResource > ¶meter) const | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inlineprotected |
| saveParameterToWriter(Serialization::SafeTensorsWriter &writer, const std::string &flat_name, const Tensor< TParameterPrecision, TMemoryResource > ¶meter, Serialization::TensorSavePass pass) const | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inlineprotected |
| setExecutionContext(IExecutionContext *context) | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inlineprotected |
| setObservedPasses(Compute::ComputePassMask passes) noexcept | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inline |
| setTrainingMode(TrainingMode mode) | Mila::Dnn::Component< TDeviceType, dtype_t::FP32 > | inline |
| synchronize() override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| TableMemoryResource typedef (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | |
| TableScaleTensorType typedef (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | |
| TableTensorType typedef (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | |
| TokenEmbedding(const std::string &name, const TokenEmbeddingConfig &config, std::optional< DeviceId > device_id=std::nullopt) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlineexplicit |
| TokenIndexType typedef (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | |
| toString() const override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| zeroGradients() override | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | inlinevirtual |
| ~TokenEmbedding() override=default (defined in Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency >) | Mila::Dnn::TokenEmbedding< TDeviceType, TIndex, TPrecision, TTableQuantization, TTableResidency > | |
1.17.0