Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision > Member List

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

AdamWOptimizer(IExecutionContext *exec_context, const AdamWConfig &config)	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	inlineexplicit
addParameter(ITensor *param, ITensor *grad) override	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	inlinevirtual
config_	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	private
context_	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	private
ExecutionContextType typedef	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >
getBeta1() const noexcept	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	inline
getBeta2() const noexcept	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	inline
getEpsilon() const noexcept	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	inline
getLearningRate() const override	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	inlinevirtual
getParameterCount() const noexcept	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	inline
getStepCount() const noexcept	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	inline
getWeightDecay() const noexcept	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	inline
impl_	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	private
OptimizerType typedef	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >
setLearningRate(float learning_rate) override	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	inlinevirtual
setWeightDecay(float weight_decay)	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	inline
step() override	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >	inlinevirtual
~AdamWOptimizer() override=default	Mila::Dnn::Optimizers::AdamWOptimizer< TDeviceType, TPrecision >
~Optimizer()=default	Mila::Dnn::Compute::Optimizer< TDeviceType, TPrecision >	virtual