Searched refs:MomentumOptimInfo (Results 1 – 3 of 3) sorted by relevance
88 class MomentumOptimInfo : public DenseOptimInfo {90 …MomentumOptimInfo(const AddressPtr &weight, const AddressPtr &accumulate, const AddressPtr &learni…92 ~MomentumOptimInfo() override = default;
265 MomentumOptimInfo::MomentumOptimInfo(const AddressPtr &weight, const AddressPtr &accumulate, in MomentumOptimInfo() function in mindspore::ps::MomentumOptimInfo280 void MomentumOptimInfo::Update(const Values &values, const Lengths &lens) { in Update()286 const AddressPtr &MomentumOptimInfo::gradient() { in gradient()293 const AddressPtr &MomentumOptimInfo::indices() { in indices()300 size_t MomentumOptimInfo::grad_index() { in grad_index()
143 return new MomentumOptimInfo(weight_addr, accumulate, learning_rate, gradient, momentum); in BuildInputs()