Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
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) overrideMila::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 typedefMila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >
getBeta1() const noexceptMila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >inline
getBeta2() const noexceptMila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >inline
getEpsilon() const noexceptMila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >inline
getLearningRate() const overrideMila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >inlinevirtual
getParameterCount() const noexceptMila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >inline
getStepCount() const noexceptMila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >inline
getWeightDecay() const noexceptMila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >inline
grad_data_Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >private
grads_Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >private
HostType typedefMila::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 typedefMila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >
param_data_Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >private
params_Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >private
setLearningRate(float learning_rate) overrideMila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >inlinevirtual
shapeToString(const shape_t &shape)Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >inlineprivatestatic
step() overrideMila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >inlinevirtual
step_count_Mila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >private
TensorType typedefMila::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=defaultMila::Dnn::Compute::CpuAdamWOptimizer< TPrecision >
~Optimizer()=defaultMila::Dnn::Compute::Optimizer< DeviceType::Cpu, TPrecision >virtual