| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/IR/ |
| H A D | TosaOps.cpp | 350 if (!inputShape.hasRank()) { in inferReturnTypeComponents()
436 if (inputShape.hasRank()) { in inferReturnTypeComponents()
500 if (!inputShape.hasRank()) { in inferReturnTypeComponents()
617 if (!inputShape.hasRank() || !inputShape.hasStaticShape()) { in inferReturnTypeComponents()
682 if (inputShape.getDimSize(0) != inputShape.getDimSize(i)) { in inferReturnTypeComponents()
691 outputShape.resize(inputShape.getRank(), inputShape.getDimSize(0)); in inferReturnTypeComponents()
747 if (inputShape.hasRank()) { in inferReturnTypeComponents()
827 if (inputShape.hasRank()) { in inferReturnTypeComponents()
979 if (!inputShape) { in NARY_SHAPE_INFER()
1028 if (inputShape.hasRank()) { in inferReturnTypeComponents()
[all …]
|
| H A D | TosaCanonicalizations.cpp | 105 ShapedType inputShape = input.getType().cast<ShapedType>(); in matchAndRewrite() local
107 inputAttr.reshape(inputShape.clone(newShapeValues)); in matchAndRewrite()
316 ArrayRef<int64_t> inputShape = inputType.getShape(); in matchAndRewrite() local
317 if (inputShape[1] != 1 || inputShape[2] != 1) { in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Tosa/Transforms/ |
| H A D | TosaDecomposeDepthwise.cpp | 60 ArrayRef<int64_t> inputShape = inputType.getShape(); in matchAndRewrite() local
62 inputShape[0], inputShape[1], inputShape[2], inputShape[3], 1}; in matchAndRewrite()
85 llvm::SmallVector<int64_t, 2> mulShape{inputShape[0], inputShape[1], in matchAndRewrite()
86 inputShape[2], inputShape[3], in matchAndRewrite()
|
| H A D | TosaDecomposeConv2D.cpp | 54 ArrayRef<int64_t> inputShape = inputType.getShape(); in matchAndRewrite() local
55 int64_t combined = inputShape[0] * inputShape[1] * inputShape[2]; in matchAndRewrite()
58 llvm::SmallVector<int64_t, 2> revisedInputShape{combined, inputShape[3]}; in matchAndRewrite()
101 llvm::SmallVector<int64_t, 4> outputShape{inputShape[0], inputShape[1], in matchAndRewrite()
102 inputShape[2], weightShape[0]}; in matchAndRewrite()
|
| H A D | TosaFoldConstantTranspose.cpp | 49 ArrayRef<int64_t> inputShape = inputType.getShape(); in matchAndRewrite() local
65 srcIndices[dim] = totalCount % inputShape[dim]; in matchAndRewrite()
66 totalCount /= inputShape[dim]; in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch3/mlir/ |
| H A D | Dialect.cpp | 294 auto inputShape = inputType.getShape(); in verify() local
295 if (!std::equal(inputShape.begin(), inputShape.end(), in verify()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch2/mlir/ |
| H A D | Dialect.cpp | 294 auto inputShape = inputType.getShape(); in verify() local
295 if (!std::equal(inputShape.begin(), inputShape.end(), in verify()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch6/mlir/ |
| H A D | Dialect.cpp | 412 auto inputShape = inputType.getShape(); in verify() local
413 if (!std::equal(inputShape.begin(), inputShape.end(), in verify()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch4/mlir/ |
| H A D | Dialect.cpp | 412 auto inputShape = inputType.getShape(); in verify() local
413 if (!std::equal(inputShape.begin(), inputShape.end(), in verify()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch5/mlir/ |
| H A D | Dialect.cpp | 412 auto inputShape = inputType.getShape(); in verify() local
413 if (!std::equal(inputShape.begin(), inputShape.end(), in verify()
|
| /llvm-project-15.0.7/mlir/examples/toy/Ch7/mlir/ |
| H A D | Dialect.cpp | 467 auto inputShape = inputType.getShape(); in verify() local
468 if (!std::equal(inputShape.begin(), inputShape.end(), in verify()
|
| /llvm-project-15.0.7/mlir/lib/Conversion/TosaToLinalg/ |
| H A D | TosaToLinalgNamed.cpp | 41 auto inputShape = inputTy.getShape(); in applyPad() local
43 assert((inputShape.size() * 2) == pad.size()); in applyPad()
48 for (int i = 0, s = inputShape.size(); i < s; i++) { in applyPad()
51 if (ShapedType::isDynamic(inputShape[i])) in applyPad()
52 paddedShape.push_back(inputShape[i]); in applyPad()
54 paddedShape.push_back(inputShape[i] + highPad + lowPad); in applyPad()
|
| H A D | TosaToLinalg.cpp | 1780 auto inputShape = inputTy.getShape(); in matchAndRewrite() local
1792 genericShape.push_back(inputShape[i]); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Math/Transforms/ |
| H A D | PolynomialApproximation.cpp | 96 ArrayRef<int64_t> inputShape = inputType.getShape(); in handleMultidimensionalVectors() local
100 if (inputShape == llvm::makeArrayRef(vectorWidth)) in handleMultidimensionalVectors()
105 int64_t innerDim = inputShape.back(); in handleMultidimensionalVectors()
111 SmallVector<int64_t> expandedShape(inputShape.begin(), inputShape.end()); in handleMultidimensionalVectors()
161 VectorType::get(inputShape, resultEltType), result); in handleMultidimensionalVectors()
|