Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision > Member List

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

addParameter(ITensor *param, ITensor *grad) override	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	inlinevirtual
config_	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	private
context_	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	private
CpuAdamWOptimizer(IExecutionContext *context, const AdamWConfig &config)	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	inlineexplicit
ExecutionContextType typedef	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >
getBeta1() const noexcept	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	inline
getBeta2() const noexcept	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	inline
getEpsilon() const noexcept	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	inline
getLearningRate() const override	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	inlinevirtual
getParameterCount() const noexcept	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	inline
getStepCount() const noexcept	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	inline
getWeightDecay() const noexcept	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	inline
grad_data_	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	private
grads_	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	private
HostType typedef	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >
learning_rate_	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	private
m_data_	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	private
m_states_	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	private
MR typedef	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >
param_data_	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	private
params_	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	private
setLearningRate(float learning_rate) override	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	inlinevirtual
shapeToString(const shape_t &shape)	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	inlineprivatestatic
step() override	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	inlinevirtual
step_count_	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	private
TensorType typedef	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >
updateParameter(HostType *param_data, const HostType *grad_data, float *m_data, float *v_data, size_t num_params, float beta1_correction, float beta2_correction)	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	inlineprivate
v_data_	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	private
v_states_	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >	private
~CpuAdamWOptimizer() override=default	Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >
~Optimizer()=default	Mila::Dnn::Compute::Optimizer< DeviceType::Cpu, TPrecision >	virtual