Searched defs:nesterov (Results 1 – 7 of 7) sorted by relevance
| /third_party/mindspore/mindspore/core/ops/ |
| D | sgd.cc | 21 void SGD::Init(const float dampening, const float weight_decay, const bool nesterov) { in Init()
34 void SGD::set_nesterov(const bool nesterov) { (void)AddAttr(kNesterov, MakeValue(nesterov)); } in set_nesterov()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | sgd_impl.cu | 21 …SGDKernel(const int size, const T dampening, const T weight_decay, const bool nesterov, const T *g… in SGDKernel()
49 void SGD(const int size, const T dampening, const T weight_decay, const bool nesterov, const T *lr,… in SGD()
|
| /third_party/mindspore/mindspore/nn/optim/ |
| D | sgd.py | 139 …t__(self, params, learning_rate=0.1, momentum=0.0, dampening=0.0, weight_decay=0.0, nesterov=False, argument
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | sgd.cc | 34 bool nesterov, int start, int end) { in DoSgd()
56 bool nesterov, int start, int end) { in DoSgdInit()
|
| D | apply_momentum.cc | 33 bool nesterov, int start, int end) { in DoApplyMomentum()
|
| D | adam.cc | 35 … float beta2_power, float eps, float learning_rate, bool nesterov, int start, int end) { in DoAdam()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 3016 def __init__(self, dampening=0.0, weight_decay=0.0, nesterov=False): argument
|