Searched refs:kApplyMomentum (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | constants.h | 50 constexpr char kApplyMomentum[] = "ApplyMomentum"; variable
239 const std::map<std::string, OptimOriginIdx> kOptimToOriginIdx = {{kApplyMomentum, kMomentumOriginId…
243 const std::map<std::string, OptimOriginIdx> kOptimToPSSendIdx = {{kApplyMomentum, kMomentumPSSendId…
|
| D | optimizer_info_builder.cc | 137 …AddressPtr learning_rate = GenInputAddrPtr<float>(kApplyMomentum, "lr", const_cast<float *>(values… in BuildInputs()
139 …AddressPtr gradient = GenInputAddrPtr<float>(kApplyMomentum, "grad", const_cast<float *>(values.da… in BuildInputs()
141 …AddressPtr momentum = GenInputAddrPtr<float>(kApplyMomentum, "momentum", const_cast<float *>(value… in BuildInputs()
|
| D | util.cc | 28 {kApplyMomentum, 0},
35 {0, kApplyMomentum},
|
| D | optimizer_info.cc | 281 UpdateOptimInputValue<float>(kApplyMomentum, "lr", const_cast<float *>(values.data()), lens); in Update()
|
| D | parameter_server.cc | 77 optim_info_builders_[kApplyMomentum] = momentum_info_builder; in InitOptimInfoBuilders()
133 } else if (optim_name == kApplyMomentum) { in InitOptimInputsShape()
|