Lines Matching refs:operandShape
377 ShapeAdaptor operandShape = operands.getShape(operand); in inferReturnTypeComponents() local
378 if (!operandShape.hasRank()) in inferReturnTypeComponents()
383 outputShape.resize(operandShape.getRank(), ShapedType::kDynamicSize); in inferReturnTypeComponents()
386 for (int i = 0, s = operandShape.getRank(); i < s; i++) { in inferReturnTypeComponents()
387 if (i == axis || operandShape.isDynamicDim(i)) in inferReturnTypeComponents()
390 outputShape[i] = operandShape.getDimSize(i); in inferReturnTypeComponents()
391 if (outputShape[i] != operandShape.getDimSize(i)) in inferReturnTypeComponents()
406 ShapeAdaptor operandShape = operands.getShape(operand); in inferReturnTypeComponents() local
410 if (!operandShape.hasRank() || operandShape.isDynamicDim(axis)) { in inferReturnTypeComponents()
415 concatDimSize += operandShape.getDimSize(axis); in inferReturnTypeComponents()
839 ShapeAdaptor operandShape, IntegerAttr axis, in ReduceInferReturnTypes() argument
841 if (!operandShape.hasRank()) { in ReduceInferReturnTypes()
847 operandShape.getDims(outputShape); in ReduceInferReturnTypes()