Searched refs:axis_idx (Results 1 – 5 of 5) sorted by relevance
111 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ResizeOutputTensor() local112 if (axis[axis_idx] == idx || axis[axis_idx] + input_num_dims == idx) { in ResizeOutputTensor()150 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ResizeOutputTensor() local151 if (axis[axis_idx] == idx || axis[axis_idx] + input_num_dims == idx) { in ResizeOutputTensor()
169 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ReducedOutputOffset() local170 if (idx == axis[axis_idx]) { in ReducedOutputOffset()551 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) {552 if (idx == axis[axis_idx]) {
388 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ReducedOutputOffset() local389 if (idx == axis[axis_idx]) { in ReducedOutputOffset()
3766 int axis_idx = node.input_size() - 1; in GetConcatAxis() local3767 while (axis_idx > 0 && IsControlInput(node.input(axis_idx))) { in GetConcatAxis()3768 --axis_idx; in GetConcatAxis()3770 if (axis_idx <= 0) { in GetConcatAxis()3774 if (!GetTensorFromConstNode(node.input(axis_idx), &axis_tensor)) { in GetConcatAxis()
949 int axis_idx = std::is_same<OpT, ConcatOp>() ? 0 : 1; in Verify() local950 Value axis = *op.getODSOperands(axis_idx).begin(); in Verify()