Searched refs:axis_idx (Results 1 – 3 of 3) sorted by relevance
105 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ResizeOutputTensor() local106 if (axis[axis_idx] == idx || axis[axis_idx] + input_num_dims == idx) { in ResizeOutputTensor()144 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ResizeOutputTensor() local145 if (axis[axis_idx] == idx || axis[axis_idx] + input_num_dims == idx) { in ResizeOutputTensor()
342 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ReducedOutputOffset() local343 if (idx == axis[axis_idx]) { in ReducedOutputOffset()
358 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ReducedOutputOffset() local359 if (idx == axis[axis_idx]) { in ReducedOutputOffset()