Searched refs:GenInputAddrPtr (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info_builder.cc | 56 AddressPtr OptimizerInfoBuilder::GenInputAddrPtr(const std::string &optim_type, const std::string &… in GenInputAddrPtr() function in mindspore::ps::OptimizerInfoBuilder
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()
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()
[all …]
|
| D | optimizer_info_builder.h | 49 …AddressPtr GenInputAddrPtr(const std::string &optim_type, const std::string &input_name, void *ps_…
|