Home
last modified time | relevance | path

Searched refs:grad_index (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/ps/
Doptimizer_info.cc41 size_t OptimizerInfo::grad_index() { return 0; } in grad_index() function in mindspore::ps::OptimizerInfo
86 size_t grad_index = this->grad_index(); in Accumulate() local
88 for (size_t i = 0; i < grad_index; i++) { in Accumulate()
94 CHECK_EQ(size, IntToSize(lengths[grad_index])); in Accumulate()
126 size_t grad_index = this->grad_index(); in Accumulate() local
128 for (size_t i = 0; i < grad_index; i++) { in Accumulate()
134 size_t incr_grad_size = IntToSize(lengths[grad_index]) * sizeof(float); in Accumulate()
146 grads_offset_ += IntToSize(lengths[grad_index]); in Accumulate()
300 size_t MomentumOptimInfo::grad_index() { in grad_index() function in mindspore::ps::MomentumOptimInfo
362 size_t SparseAdamOptimInfo::grad_index() { in grad_index() function in mindspore::ps::SparseAdamOptimInfo
[all …]
Doptimizer_info.h48 virtual size_t grad_index();
97 size_t grad_index() override;
112 size_t grad_index() override;
125 size_t grad_index() override;
Dworker.cc68 int64_t grad_index = -1; in Push() local
73 grad_index = kSparseGradIndex; in Push()
78 grad_index = 0; in Push()
115 << " the grad_index:" << grad_index << " the indice_index:" << indice_index in Push()
117 …arseData(std::vector<Key>(keys), total_buffer, std::vector<int>(sizes_int), LongToSize(grad_index), in Push()
569 void Worker::BuildSparseValue(const std::vector<int> &lengths, const size_t grad_index, const size_… in BuildSparseValue() argument
582 if (i != grad_index && i != indice_index) { in BuildSparseValue()
601 for (size_t i = 0; i < grad_index; i++) { in BuildSparseValue()
604 size_t data_size = lengths[grad_index] * sizeof(float); in BuildSparseValue()
617 int64_t indice_offset = grad_offset + lengths[grad_index]; in BuildSparseValue()
[all …]
Dworker.h101 …void BuildSparseValue(const std::vector<int> &lengths, const size_t grad_index, const size_t indic…
107 … size_t grad_index, size_t indice_index, size_t first_dim_size, size_t outer_dim_size);