Mila 0.13.48
Deep Neural Network Library
Loading...
Searching...
No Matches
Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision > Member List

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

addParameter(ITensor *param, ITensor *grad) overrideMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >inlinevirtual
config_Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >private
CudaAdamWOptimizer(IExecutionContext *context, const OptimizerConfig &config)Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >inlineexplicit
CudaExecutionContext typedefMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >
exec_context_Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >private
getBeta1() const noexceptMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >inline
getBeta2() const noexceptMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >inline
getEpsilon() const noexceptMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >inline
getLearningRate() const overrideMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >inlinevirtual
getParameterCount() const noexceptMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >inline
getStepCount() const noexceptMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >inline
getWeightDecay() const noexceptMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >inline
grad_data_Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >private
grad_scale_Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >private
grads_Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >private
m_data_Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >private
m_states_Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >private
master_param_data_Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >private
master_params_Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >private
MR typedefMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >
NativeType typedefMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >
param_data_Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >private
params_Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >private
setLearningRate(float learning_rate) overrideMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >inlinevirtual
setWeightDecay(float weight_decay)Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >inline
shapeToString(const shape_t &shape)Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >inlineprivatestatic
step() overrideMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >inlinevirtual
step_count_Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >private
TensorType typedefMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >
v_data_Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >private
v_states_Mila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >private
validateHyperparameters() constMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >inlineprivate
~CudaAdamWOptimizer() override=defaultMila::Dnn::Compute::CudaAdamWOptimizer< TPrecision >
~Optimizer()=defaultMila::Dnn::Compute::Optimizer< DeviceType::Cuda, TPrecision >virtual