| awaitSampledToken() | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | inlineprotected |
| Base typedef (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | |
| contextLength() const noexcept | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | inline |
| enqueueSampleNext(const TensorType &logits, TokenTensor &token_out, const SamplingParams ¶ms) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | inlineprotected |
| fingerprintPrefill(const std::vector< int32_t > &token_ids) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | inline |
| fromPretrained(const std::filesystem::path &path, const GemmaModelConfig &model_config, DeviceId device_id=DeviceId{ TDeviceType, 0 }) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | inlinestatic |
| GemmaModel(const GemmaModel &)=delete (defined in Mila::Dnn::GemmaModel< TDeviceType, TPrecision >) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | |
| GemmaModel(GemmaModel &&)=default (defined in Mila::Dnn::GemmaModel< TDeviceType, TPrecision >) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | |
| GemmaSlidingKvPolicy typedef | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | |
| generate(std::span< const int32_t > prompt_tokens, const std::function< void(int32_t)> &on_token, const GenerateParams ¶ms={}, std::stop_token stop={}) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | inline |
| getDeviceId() const noexcept | Mila::Dnn::Model< TDeviceType, TPrecision > | inline |
| getLanguageNetwork() noexcept (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | inlineprotected |
| getLanguageNetwork() const noexcept (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | inlineprotected |
| getMemoryStats() const | Mila::Dnn::Model< TDeviceType, TPrecision > | inline |
| getModelConfig() const noexcept | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | inline |
| getNetworkConfig() const noexcept | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | inline |
| getRequiredMemory(const std::filesystem::path &path, const GemmaModelConfig &model_config, DeviceId device_id=DeviceId{ TDeviceType, 0 }) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | inlinestatic |
| getRuntimeMode() const noexcept | Mila::Dnn::Model< TDeviceType, TPrecision > | inline |
| getScratchHighWaterBytes() const | Mila::Dnn::Model< TDeviceType, TPrecision > | inline |
| isEval() const noexcept | Mila::Dnn::Model< TDeviceType, TPrecision > | inline |
| isInferenceMode() const noexcept | Mila::Dnn::Model< TDeviceType, TPrecision > | inline |
| isTrainingMode() const noexcept | Mila::Dnn::Model< TDeviceType, TPrecision > | inline |
| LanguageModel(const LanguageModel &)=delete (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | |
| LanguageModel(LanguageModel &&)=default (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | |
| LanguageModel(std::unique_ptr< LanguageNetwork< TDeviceType, TPrecision > > network, RuntimeMode runtime_mode, Serialization::PretrainedMetadata source_metadata={}, WeightQuantization weight_quantization=WeightQuantization::None) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | inlineexplicitprotected |
| maxSequenceLength() const noexcept override (defined in Mila::Dnn::GemmaModel< TDeviceType, TPrecision >) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | inlineprotectedvirtual |
| Model(const Model &)=delete (defined in Mila::Dnn::Model< TDeviceType, TPrecision >) | Mila::Dnn::Model< TDeviceType, TPrecision > | |
| Model(Model &&)=default (defined in Mila::Dnn::Model< TDeviceType, TPrecision >) | Mila::Dnn::Model< TDeviceType, TPrecision > | |
| Model(std::unique_ptr< NetworkType > network, RuntimeMode runtime_mode) | Mila::Dnn::Model< TDeviceType, TPrecision > | inlineexplicitprotected |
| ModelBase typedef (defined in Mila::Dnn::GemmaModel< TDeviceType, TPrecision >) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | |
| MR typedef (defined in Mila::Dnn::GemmaModel< TDeviceType, TPrecision >) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | |
| network_ | Mila::Dnn::Model< TDeviceType, TPrecision > | protected |
| NetworkType typedef (defined in Mila::Dnn::Model< TDeviceType, TPrecision >) | Mila::Dnn::Model< TDeviceType, TPrecision > | |
| onGenerating(std::span< const int32_t > prompt_tokens, const std::function< void(int32_t)> &on_token, const GenerateParams ¶ms, std::stop_token stop) override | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | inlineprotectedvirtual |
| onTraining() override | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | inlineprotectedvirtual |
| operator=(const GemmaModel &)=delete (defined in Mila::Dnn::GemmaModel< TDeviceType, TPrecision >) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | |
| operator=(GemmaModel &&)=default (defined in Mila::Dnn::GemmaModel< TDeviceType, TPrecision >) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | |
| operator=(const LanguageModel &)=delete (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | |
| operator=(LanguageModel &&)=default (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | |
| operator=(const Model &)=delete (defined in Mila::Dnn::Model< TDeviceType, TPrecision >) | Mila::Dnn::Model< TDeviceType, TPrecision > | |
| operator=(Model &&)=default (defined in Mila::Dnn::Model< TDeviceType, TPrecision >) | Mila::Dnn::Model< TDeviceType, TPrecision > | |
| profilePrefill(const std::vector< int32_t > &token_ids) (defined in Mila::Dnn::GemmaModel< TDeviceType, TPrecision >) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | inline |
| sampleNext(const TensorType &logits, TokenTensor &token_out, const SamplingParams ¶ms) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | inlineprotected |
| savePretrained(const std::filesystem::path &path) const | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | inline |
| seedSampler(uint64_t seed) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | inline |
| setEval(bool eval) | Mila::Dnn::Model< TDeviceType, TPrecision > | inline |
| source_metadata_ | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | protected |
| StagingMR typedef (defined in Mila::Dnn::GemmaModel< TDeviceType, TPrecision >) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | |
| TensorType typedef (defined in Mila::Dnn::GemmaModel< TDeviceType, TPrecision >) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | |
| TokenIndexType typedef (defined in Mila::Dnn::GemmaModel< TDeviceType, TPrecision >) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | |
| TokenTensor typedef (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | protected |
| toString() const override | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | inlinevirtual |
| train() | Mila::Dnn::Model< TDeviceType, TPrecision > | inline |
| vocabSize() const noexcept override (defined in Mila::Dnn::GemmaModel< TDeviceType, TPrecision >) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | inlineprotectedvirtual |
| weight_quantization_ | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | protected |
| ~GemmaModel()=default (defined in Mila::Dnn::GemmaModel< TDeviceType, TPrecision >) | Mila::Dnn::GemmaModel< TDeviceType, TPrecision > | |
| ~LanguageModel()=default (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | virtual |
| ~Model()=default (defined in Mila::Dnn::Model< TDeviceType, TPrecision >) | Mila::Dnn::Model< TDeviceType, TPrecision > | virtual |
1.17.0