Mila::Dnn::LanguageModel< TDeviceType, TPrecision > Member List

This is the complete list of members for Mila::Dnn::LanguageModel< TDeviceType, TPrecision >, including all inherited members.

Base typedef	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >
ensureTrainingMode(const char *method) const	Mila::Dnn::Model< TDeviceType, TPrecision >	inlineprivate
eosToken() const noexcept=0	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >	protectedpure virtual
generate(const std::vector< int32_t > &prompt_tokens, size_t max_new_tokens=64, float temperature=1.0f, int top_k=0)	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >	inline
generateStreaming(const std::vector< int32_t > &prompt_tokens, std::function< void(int32_t)> on_token, size_t max_new_tokens=64, float temperature=1.0f, int top_k=0, std::stop_token stop={})	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >	inline
getDeviceId() const noexcept	Mila::Dnn::Model< TDeviceType, TPrecision >	inline
getLanguageNetwork() noexcept	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >	inlineprotected
getLanguageNetwork() const noexcept	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >	inlineprotected
getLastGenerationStatistics() const noexcept	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >	inline
getMemoryStats() const	Mila::Dnn::Model< TDeviceType, TPrecision >	inline
getRuntimeMode() const noexcept	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	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >
LanguageModel(LanguageModel &&)=default	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >
LanguageModel(std::unique_ptr< LanguageNetwork< TDeviceType, TPrecision > > network, RuntimeMode runtime_mode)	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >	inlineexplicitprotected
last_generation_statistics_	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >	protected
maxSequenceLength() const noexcept=0	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >	protectedpure virtual
Model(const Model &)=delete	Mila::Dnn::Model< TDeviceType, TPrecision >
Model(Model &&)=default	Mila::Dnn::Model< TDeviceType, TPrecision >
Model(std::unique_ptr< NetworkType > network, RuntimeMode runtime_mode)	Mila::Dnn::Model< TDeviceType, TPrecision >	inlineexplicitprotected
network_	Mila::Dnn::Model< TDeviceType, TPrecision >	protected
NetworkType typedef	Mila::Dnn::Model< TDeviceType, TPrecision >
onGenerating(const std::vector< int32_t > &prompt_tokens, const std::function< void(int32_t)> &on_token, size_t max_new_tokens, float temperature, int top_k, 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	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >
operator=(LanguageModel &&)=default	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >
Mila::Dnn::Model::operator=(const Model &)=delete	Mila::Dnn::Model< TDeviceType, TPrecision >
Mila::Dnn::Model::operator=(Model &&)=default	Mila::Dnn::Model< TDeviceType, TPrecision >
runtime_mode_	Mila::Dnn::Model< TDeviceType, TPrecision >	private
setEval(bool eval)	Mila::Dnn::Model< TDeviceType, TPrecision >	inline
stopTokens() const	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >	inlineprotectedvirtual
toString() const =0	Mila::Dnn::Model< TDeviceType, TPrecision >	pure virtual
train()	Mila::Dnn::Model< TDeviceType, TPrecision >	inline
vocabSize() const noexcept=0	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >	protectedpure virtual
~LanguageModel()=default	Mila::Dnn::LanguageModel< TDeviceType, TPrecision >	virtual
~Model()=default	Mila::Dnn::Model< TDeviceType, TPrecision >	virtual