Searched defs:dampening (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()
27 void SGD::set_dampening(const float dampening) { in set_dampening()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/gpu/cuda_impl/ |
| D | sgd_impl.cu | 21 __global__ void SGDKernel(const int size, const T dampening, const T weight_decay, const bool neste… 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 …def __init__(self, params, learning_rate=0.1, momentum=0.0, dampening=0.0, weight_decay=0.0, neste… argument
|
| /third_party/mindspore/mindspore/lite/src/runtime/kernel/arm/fp32_grad/ |
| D | sgd.cc | 33 …ight, float *accumulate, const float *gradient, float learning_rate, float dampening, float moment, in DoSgd()
|
| /third_party/mindspore/mindspore/ops/operations/ |
| D | nn_ops.py | 3016 def __init__(self, dampening=0.0, weight_decay=0.0, nesterov=False): argument
|
| /third_party/mindspore/mindspore/lite/schema/ |
| D | ops_v0_generated.h | 9895 float dampening() const { in dampening() function
|
| /third_party/mindspore/mindspore/lite/schema/inner/ |
| D | ops_v0_generated.h | 13914 float dampening = 0.0f; member
13935 float dampening() const { in dampening() function
|