Searched refs:kSparseAdam (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info_builder.cc | 184 …AddressPtr beta1_power = GenInputAddrPtr<float>(kSparseAdam, "beta1_power", const_cast<float *>(va… in BuildInputs()
186 …AddressPtr beta2_power = GenInputAddrPtr<float>(kSparseAdam, "beta2_power", const_cast<float *>(va… in BuildInputs()
188 …AddressPtr learning_rate = GenInputAddrPtr<float>(kSparseAdam, "lr", const_cast<float *>(values.da… in BuildInputs()
190 …AddressPtr beta1 = GenInputAddrPtr<float>(kSparseAdam, "beta1", const_cast<float *>(values.data())… in BuildInputs()
192 …AddressPtr beta2 = GenInputAddrPtr<float>(kSparseAdam, "beta2", const_cast<float *>(values.data())… in BuildInputs()
194 …AddressPtr epsilon = GenInputAddrPtr<float>(kSparseAdam, "eps", const_cast<float *>(values.data())… in BuildInputs()
196 …AddressPtr grad = GenInputAddrPtr<float>(kSparseAdam, "grad", const_cast<float *>(values.data()), … in BuildInputs()
199 …GenInputAddrPtr<float>(kSparseAdam, "indices", const_cast<float *>(values.data()), lens, inputs_sh… in BuildInputs()
|
| D | optimizer_info.cc | 338 …UpdateOptimInputValue<float>(kSparseAdam, "beta1_power", const_cast<float *>(values.data()), lens); in Update()
339 …UpdateOptimInputValue<float>(kSparseAdam, "beta2_power", const_cast<float *>(values.data()), lens); in Update()
340 UpdateOptimInputValue<float>(kSparseAdam, "lr", const_cast<float *>(values.data()), lens); in Update()
341 UpdateOptimInputValue<float>(kSparseAdam, "beta1", const_cast<float *>(values.data()), lens); in Update()
342 UpdateOptimInputValue<float>(kSparseAdam, "beta2", const_cast<float *>(values.data()), lens); in Update()
343 UpdateOptimInputValue<float>(kSparseAdam, "eps", const_cast<float *>(values.data()), lens); in Update()
|
| D | constants.h | 51 constexpr char kSparseAdam[] = "Adam"; variable
240 … {kSparseAdam, kSparseAdamOriginIdx},
244 … {kSparseAdam, kSparseAdamPSSendIdx},
|
| D | util.cc | 29 {kSparseAdam, 1},
36 {1, kSparseAdam},
|
| D | parameter_server.cc | 78 optim_info_builders_[kSparseAdam] = sparse_adam_info_builder; in InitOptimInfoBuilders()
123 if (optim_name == kSparseAdam) { in InitOptimInputsShape()
|