Lines Matching defs:inputShape
633 SmallVector<OpFoldResult> inputShape =
636 outputShape = inputShape;
642 {outputShape[concatDim], inputShape[concatDim]});
644 inputShapes.emplace_back(std::move(inputShape));
1726 ArrayRef<OpFoldResult> inputShape) {
1729 inputShape);
1753 SmallVector<OpFoldResult> inputShape =
1757 builder, result.location, tensorResultTy, reassociation, inputShape);
4140 bool PackOp::requirePaddingValue(ArrayRef<int64_t> inputShape,
4146 outputShape.take_front(inputShape.size()));
4154 if (ShapedType::isDynamic(inputShape[pos]))
4160 (inputShape[pos] % outputTileSizes[pos] != 0))
4162 } else if (inputShape[pos] % (*constantTile) != 0) {