Home
last modified time | relevance | path

Searched refs:ori_index (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/lite/src/runtime/kernel/opencl/kernel/
Dconv2d_transpose.cc202 int ori_index = ((ci_offset * kh + kh_i) * kw + kw_i) * co + co_offset; in InitFilter() local
206 reinterpret_cast<float *>(origin_weight)[ori_index]; in InitFilter()
209 reinterpret_cast<float16_t *>(origin_weight)[ori_index]; in InitFilter()
214 reinterpret_cast<float *>(origin_weight)[ori_index]; in InitFilter()
217 reinterpret_cast<float16_t *>(origin_weight)[ori_index]; in InitFilter()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/common/
Dhelper.cc737 size_t ori_index = 0; in RectifyAbstractFromRegAttr() local
750 rectify_abs = input_abstract[ori_index++]; in RectifyAbstractFromRegAttr()
755 if (ori_index > input_abstract.size()) { in RectifyAbstractFromRegAttr()
757 << " get index :" << ori_index; in RectifyAbstractFromRegAttr()
759 rectify_abs_list[index] = input_abstract[ori_index++]; in RectifyAbstractFromRegAttr()
/third_party/mindspore/mindspore/ccsrc/backend/optimizer/somas/
Dsomas.cc819 auto ori_index = origin_pair.second; in RefNodeProcess() local
827 auto input_tensor = repeat_node->output_tensors_[ori_index]; in RefNodeProcess()