Lines Matching defs:axis
445 int32_t axis = concatOp.getAxis();
463 // Validate slice on the concatenated axis. Slicing along this
464 // axis should span only one of the inputs to the concatenate
473 if (sliceStarts[axis] >= 0 && (sliceStarts[axis] + sliceSizes[axis]) <=
474 inputType.getDimSize(axis)) {
486 sliceStarts[axis] -= inputType.getDimSize(axis);
1005 auto axis = getAxis();
1013 (operandTy.getRank() == 0 || operandTy.getDimSize(axis) == 1))
1153 // Fold consecutive concats on the same axis into a single op.