Searched refs:WeightPtr (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info_builder.h | 36 …OptimizerInfo *Build(const std::shared_ptr<PServerKernel> &pserver_kernel, const WeightPtr &weight…
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 | parameter_server.h | 125 void InitWeight(const Key &key, const WeightPtr &weight);
134 WeightPtr weight(const Key &key);
166 std::unordered_map<Key, WeightPtr> weights_;
168 std::unordered_map<Key, WeightPtr> grads_;
|
| D | optimizer_info_builder.cc | 27 … const WeightPtr &weight, const Keys &keys, const Values &values, in Build()
114 OptimizerInfo *MomentumOptimInfoBuilder::BuildInputs(const WeightPtr &weight, const Keys &, const V… in BuildInputs()
146 OptimizerInfo *SparseAdamOptimInfoBuilder::BuildInputs(const WeightPtr &weight, const Keys &, const… in BuildInputs()
205 OptimizerInfo *SparseFtrlOptimInfoBuilder::BuildInputs(const WeightPtr &weight, const Keys &, const… in BuildInputs()
|
| D | parameter_server.cc | 148 void ParameterServer::InitWeight(const Key &key, const WeightPtr &weight) { in InitWeight()
230 WeightPtr embedding = std::make_shared<Weight>(total_dims, 0); in InitEmbeddingTable()
283 WeightPtr weight_ptr = iter->second; in UpdateWeights()
356 WeightPtr ParameterServer::weight(const Key &key) { in weight()
361 WeightPtr weight_ptr = weights_[key]; in weight()
363 WeightPtr copy_weight_ptr = std::make_shared<std::vector<float>>(weight_ptr->size(), 0); in weight()
381 WeightPtr table_ptr = weights_[key]; in DoEmbeddingLookup()
437 WeightPtr table_ptr = weights_[key]; in UpdateEmbeddings()
669 …WeightPtr weight_ptr = std::make_shared<std::vector<float>>(data_ptr + pos, data_ptr + (pos + data… in HandleInitWeights()
|
| D | constants.h | 205 using WeightPtr = std::shared_ptr<Weight>; variable
|