Searched defs:grad_index (Results 1 – 2 of 2) sorted by relevance
41 size_t OptimizerInfo::grad_index() { return 0; } in grad_index() function in mindspore::ps::OptimizerInfo86 size_t grad_index = this->grad_index(); in Accumulate() local126 size_t grad_index = this->grad_index(); in Accumulate() local300 size_t MomentumOptimInfo::grad_index() { in grad_index() function in mindspore::ps::MomentumOptimInfo362 size_t SparseAdamOptimInfo::grad_index() { in grad_index() function in mindspore::ps::SparseAdamOptimInfo405 size_t SparseFtrlOptimInfo::grad_index() { in grad_index() function in mindspore::ps::SparseFtrlOptimInfo
68 int64_t grad_index = -1; in Push() local569 void Worker::BuildSparseValue(const std::vector<int> &lengths, const size_t grad_index, const size_… in BuildSparseValue()660 … size_t grad_index, size_t indice_index, size_t first_dim_size, size_t outer_dim_size) { in PushSparseData()733 size_t grad_index = static_cast<size_t>(iter->second); in SparsePartitioner() local