Searched refs:SparseAdamOptimInfo (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | optimizer_info.cc | 305 SparseAdamOptimInfo::SparseAdamOptimInfo(const AddressPtr &weight, const AddressPtr &m, const Addre… in SparseAdamOptimInfo() function in mindspore::ps::SparseAdamOptimInfo
337 void SparseAdamOptimInfo::Update(const Values &values, const Lengths &lens) { in Update()
346 const AddressPtr &SparseAdamOptimInfo::gradient() { in gradient()
353 const AddressPtr &SparseAdamOptimInfo::indices() { in indices()
360 bool SparseAdamOptimInfo::IsSparse() const { return true; } in IsSparse()
362 size_t SparseAdamOptimInfo::grad_index() { in grad_index()
367 size_t SparseAdamOptimInfo::indices_index() { in indices_index()
|
| D | optimizer_info.h | 100 class SparseAdamOptimInfo : public SparseOptimInfo {
102 …SparseAdamOptimInfo(const AddressPtr &weight, const AddressPtr &m, const AddressPtr &v, const Addr…
106 ~SparseAdamOptimInfo() override = default;
|
| D | optimizer_info_builder.cc | 201 …return new SparseAdamOptimInfo(weight_addr, m, v, beta1_power, beta2_power, learning_rate, beta1, … in BuildInputs()
|