Searched refs:BuildInputs (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info_builder.h | 40 …virtual OptimizerInfo *BuildInputs(const WeightPtr &weight, const Keys &keys, const Values &values,
59 …OptimizerInfo *BuildInputs(const WeightPtr &weight, const Keys &keys, const Values &values, const …
68 …OptimizerInfo *BuildInputs(const WeightPtr &weight, const Keys &keys, const Values &values, const …
77 …OptimizerInfo *BuildInputs(const WeightPtr &weight, const Keys &keys, const Values &values, const …
|
| D | optimizer_info_builder.cc | 34 BuildInputs(weight, keys, values, lens, inputs_shape, worker_num, pserver_kernel, sharded); in Build()
114 OptimizerInfo *MomentumOptimInfoBuilder::BuildInputs(const WeightPtr &weight, const Keys &, const V… in BuildInputs() function in mindspore::ps::MomentumOptimInfoBuilder
146 OptimizerInfo *SparseAdamOptimInfoBuilder::BuildInputs(const WeightPtr &weight, const Keys &, const… in BuildInputs() function in mindspore::ps::SparseAdamOptimInfoBuilder
205 OptimizerInfo *SparseFtrlOptimInfoBuilder::BuildInputs(const WeightPtr &weight, const Keys &, const… in BuildInputs() function in mindspore::ps::SparseFtrlOptimInfoBuilder
|