Searched refs:transposeVector (Results 1 – 6 of 6) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Transforms/ |
| H A D | HoistPadding.cpp | 151 ArrayRef<int64_t> transposeVector) { in computeTransposedType() argument
152 if (transposeVector.empty()) in computeTransposedType()
154 if (!isPermutation(transposeVector) || in computeTransposedType()
155 transposeVector.size() != static_cast<size_t>(rankedTensorType.getRank())) in computeTransposedType()
160 applyPermutationToVector(transposedShape, transposeVector); in computeTransposedType()
398 tensor::PadOp opToHoist, int numLoops, ArrayRef<int64_t> transposeVector, in hoistPaddingOnTensors() argument
422 computeTransposedType(paddedTensorType, transposeVector); in hoistPaddingOnTensors()
503 if (!transposeVector.empty()) { in hoistPaddingOnTensors()
507 makeTransposeOp(b, loc, paddedTensor, outputTensor, transposeVector)); in hoistPaddingOnTensors()
545 if (!transposeVector.empty()) { in hoistPaddingOnTensors()
[all …]
|
| H A D | Transforms.cpp | 435 SmallVector<int64_t> transposeVector = in returningMatchAndRewrite()
441 padOp, en.value(), transposeVector, hoistedOp, transposeOps); in returningMatchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Transforms/ |
| H A D | HoistPadding.h | 71 tensor::PadOp opToHoist, int numLoops, ArrayRef<int64_t> transposeVector,
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/Utils/ |
| H A D | Utils.cpp | 438 ArrayRef<int64_t> transposeVector) { in makeTransposeOp() argument
442 assert(isPermutation(transposeVector) && in makeTransposeOp()
444 assert(transposeVector.size() == in makeTransposeOp()
451 SmallVector<unsigned>(transposeVector.begin(), transposeVector.end()), in makeTransposeOp()
453 AffineMap::getMultiDimIdentityMap(transposeVector.size(), in makeTransposeOp()
455 SmallVector<llvm::StringRef> iteratorTypes(transposeVector.size(), in makeTransposeOp()
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Linalg/Utils/ |
| H A D | Utils.h | 139 ArrayRef<int64_t> transposeVector);
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Linalg/TransformOps/ |
| H A D | LinalgTransformOps.cpp | 559 for (Attribute transposeVector : getTransposePaddings().cast<ArrayAttr>()) in applyToOne()
561 extractFromI64ArrayAttr(transposeVector.cast<ArrayAttr>())); in applyToOne()
|