Searched refs:getShapeForUnroll (Results 1 – 4 of 4) sorted by relevance
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/Transforms/ |
| H A D | VectorUnrollDistribute.cpp | 152 auto maybeUnrollShape = unrollableVectorOp.getShapeForUnroll(); in getTargetShape()
319 SmallVector<int64_t, 4> originalSize = *contractOp.getShapeForUnroll(); in matchAndRewrite()
421 SmallVector<int64_t, 4> originalSize = *reductionOp.getShapeForUnroll(); in matchAndRewrite()
496 *cast<VectorUnrollOpInterface>(op).getShapeForUnroll(); in matchAndRewrite()
739 SmallVector<int64_t> originalSize = *reductionOp.getShapeForUnroll(); in matchAndRewrite()
|
| /llvm-project-15.0.7/mlir/include/mlir/Interfaces/ |
| H A D | VectorInterfaces.td | 32 /*methodName=*/"getShapeForUnroll",
|
| /llvm-project-15.0.7/mlir/lib/Dialect/Vector/IR/ |
| H A D | VectorOps.cpp | 354 Optional<SmallVector<int64_t, 4>> MultiDimReductionOp::getShapeForUnroll() { in getShapeForUnroll() function in MultiDimReductionOp
486 Optional<SmallVector<int64_t, 4>> ReductionOp::getShapeForUnroll() { in getShapeForUnroll() function in ReductionOp
883 Optional<SmallVector<int64_t, 4>> ContractionOp::getShapeForUnroll() { in getShapeForUnroll() function in ContractionOp
1676 Optional<SmallVector<int64_t, 4>> FMAOp::getShapeForUnroll() { in getShapeForUnroll() function in FMAOp
3184 Optional<SmallVector<int64_t, 4>> TransferReadOp::getShapeForUnroll() { in getShapeForUnroll() function in TransferReadOp
3622 Optional<SmallVector<int64_t, 4>> TransferWriteOp::getShapeForUnroll() { in getShapeForUnroll() function in TransferWriteOp
4610 Optional<SmallVector<int64_t, 4>> TransposeOp::getShapeForUnroll() { in getShapeForUnroll() function in TransposeOp
|
| /llvm-project-15.0.7/mlir/include/mlir/Dialect/Vector/IR/ |
| H A D | VectorOps.td | 80 DeclareOpInterfaceMethods<VectorUnrollOpInterface, ["getShapeForUnroll"]>
270 ["getShapeForUnroll"]>]>,
321 ["getShapeForUnroll"]>]>,
650 DeclareOpInterfaceMethods<VectorUnrollOpInterface, ["getShapeForUnroll"]>
1160 DeclareOpInterfaceMethods<VectorUnrollOpInterface, ["getShapeForUnroll"]>,
1366 DeclareOpInterfaceMethods<VectorUnrollOpInterface, ["getShapeForUnroll"]>,
2225 DeclareOpInterfaceMethods<VectorUnrollOpInterface, ["getShapeForUnroll"]>,
|