Searched defs:axis_idx (Results 1 – 5 of 5) sorted by relevance
111 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ResizeOutputTensor() local150 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ResizeOutputTensor() local
169 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ReducedOutputOffset() local
388 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ReducedOutputOffset() local
949 int axis_idx = std::is_same<OpT, ConcatOp>() ? 0 : 1; in Verify() local
3766 int axis_idx = node.input_size() - 1; in GetConcatAxis() local