| awaitSampledToken() | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | inlineprotected |
| Base typedef (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | |
| enqueueSampleNext(const TensorType &logits, TokenTensor &token_out, const SamplingParams ¶ms) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | inlineprotected |
| eosToken() const noexcept=0 (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | protectedpure virtual |
| finalLogitSoftcap() const noexcept | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | inlineprotectedvirtual |
| 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 |
| 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=0 (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | protectedpure virtual |
| 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 |
| MR typedef (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | protected |
| 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)=0 | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | protectedpure virtual |
| onTraining()=0 | Mila::Dnn::Model< TDeviceType, TPrecision > | protectedpure virtual |
| 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 > | |
| 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 |
| stopTokens() const (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | inlineprotectedvirtual |
| TensorType typedef (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | protected |
| TokenTensor typedef (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | protected |
| toString() const =0 | Mila::Dnn::Model< TDeviceType, TPrecision > | pure virtual |
| train() | Mila::Dnn::Model< TDeviceType, TPrecision > | inline |
| vocabSize() const noexcept=0 (defined in Mila::Dnn::LanguageModel< TDeviceType, TPrecision >) | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | protectedpure virtual |
| weight_quantization_ | Mila::Dnn::LanguageModel< TDeviceType, TPrecision > | protected |
| ~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