Searched refs:kSparseFtrl (Results 1 – 4 of 4) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | constants.h | 53 constexpr char kSparseFtrl[] = "Ftrl"; variable
242 … {kSparseFtrl, kSparseFtrlOriginIdx}};
246 … {kSparseFtrl, kSparseFtrlPSSendIdx}};
|
| D | util.cc | 31 {kSparseFtrl, 3},
38 {3, kSparseFtrl},
|
| D | optimizer_info_builder.cc | 240 …AddressPtr grad = GenInputAddrPtr<float>(kSparseFtrl, "grad", const_cast<float *>(values.data()), … in BuildInputs()
243 …GenInputAddrPtr<float>(kSparseFtrl, "indices", const_cast<float *>(values.data()), lens, inputs_sh… in BuildInputs()
|
| D | parameter_server.cc | 79 optim_info_builders_[kSparseFtrl] = sparse_ftrl_info_builder; in InitOptimInfoBuilders()
138 } else if (optim_name == kSparseFtrl) { in InitOptimInputsShape()
|