Searched refs:val_idx (Results 1 – 5 of 5) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp32_grad/ |
| D | pooling_grad.c | 144 int val_idx = (xw + in_w * xh) * channel + ic; in MaxPoolingGrad() local
146 uint32x4_t index = {val_idx, val_idx + 1, val_idx + 2, val_idx + 3}; in MaxPoolingGrad()
147 float32x4_t in = vld1q_f32(inPtr + val_idx); in MaxPoolingGrad()
151 … float val[4] = {inPtr[val_idx], inPtr[val_idx + 1], inPtr[val_idx + 2], inPtr[val_idx + 3]}; in MaxPoolingGrad()
155 max_idx[i] = val_idx + i; in MaxPoolingGrad()
174 int val_idx = (xw + in_w * xh) * channel + ic; in MaxPoolingGrad() local
175 float val = inPtr[val_idx]; in MaxPoolingGrad()
178 max_idx = val_idx; in MaxPoolingGrad()
|
| /third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/fp16_grad/ |
| D | pooling_grad.c | 147 int val_idx = (xw + in_w * xh) * channel + ic; in MaxPoolingFp16Grad() local
149 uint32x4_t index = {val_idx, val_idx + 1, val_idx + 2, val_idx + 3}; in MaxPoolingFp16Grad()
150 float16x4_t in = vld1_f16(inPtr + val_idx); in MaxPoolingFp16Grad()
153 … float16_t val[4] = {inPtr[val_idx], inPtr[val_idx + 1], inPtr[val_idx + 2], inPtr[val_idx + 3]}; in MaxPoolingFp16Grad()
157 max_idx[i] = val_idx + i; in MaxPoolingFp16Grad()
176 int val_idx = (xw + in_w * xh) * channel + ic; in MaxPoolingFp16Grad() local
177 float16_t val = inPtr[val_idx]; in MaxPoolingFp16Grad()
180 max_idx = val_idx; in MaxPoolingFp16Grad()
|
| /third_party/mindspore/mindspore/ccsrc/minddata/dataset/engine/datasetops/source/sampler/ |
| D | weighted_random_sampler.cc | 85 std::vector<std::pair<double, int64_t>> val_idx; in InitOnePassSampling() local
87 val_idx.emplace_back(std::make_pair((*exp_dist_)(rand_gen_) / weights_[i], i)); in InitOnePassSampling()
91 std::partial_sort(val_idx.begin(), val_idx.begin() + num_samples_, val_idx.end()); in InitOnePassSampling()
93 onepass_ids_.push_back(val_idx[i].second); in InitOnePassSampling()
|
| /third_party/python/Python/ |
| D | hamt.c | 597 uint32_t val_idx = key_idx + 1; in hamt_node_bitmap_clone_without() local
606 for (i = val_idx + 1; i < (uint32_t)Py_SIZE(o); i++) { in hamt_node_bitmap_clone_without()
712 uint32_t val_idx = key_idx + 1; in hamt_node_bitmap_assoc() local
714 assert(val_idx < (size_t)Py_SIZE(self)); in hamt_node_bitmap_assoc()
717 PyObject *val_or_node = self->b_array[val_idx]; in hamt_node_bitmap_assoc()
742 Py_SETREF(ret->b_array[val_idx], (PyObject*)sub_node); in hamt_node_bitmap_assoc()
768 Py_SETREF(ret->b_array[val_idx], val); in hamt_node_bitmap_assoc()
796 Py_SETREF(ret->b_array[val_idx], (PyObject *)sub_node); in hamt_node_bitmap_assoc()
905 uint32_t val_idx = key_idx + 1; in hamt_node_bitmap_assoc() local
929 new_node->b_array[val_idx] = val; in hamt_node_bitmap_assoc()
[all …]
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | parameter_server.cc | 97 size_t val_idx = 0; in InitOptimInputsShape() local
113 shape->push_back(values[val_idx]); in InitOptimInputsShape()
114 original_shape->push_back(values[val_idx++]); in InitOptimInputsShape()
|