Lines Matching defs:outputShape

626   SmallVector<OpFoldResult> outputShape;
636       outputShape = inputShape;
639       concatOffsets.push_back(outputShape[concatDim]);
640       outputShape[concatDim] = affine::makeComposedFoldedAffineApply(
642           {outputShape[concatDim], inputShape[concatDim]});
648       loc, outputShape, getType().getElementType());
1727   std::optional<SmallVector<OpFoldResult>> outputShape =
1730   if (!outputShape)
1732   return *outputShape;
1742                           ArrayRef<OpFoldResult> outputShape) {
1744       decomposeMixedValues(SmallVector<OpFoldResult>(outputShape));
1756   FailureOr<SmallVector<OpFoldResult>> outputShape = inferOutputShape(
1759   if (succeeded(outputShape)) {
1760     outputShapeOrEmpty = *outputShape;
4142                                  ArrayRef<int64_t> outputShape,
4146       outputShape.take_front(inputShape.size()));