Searched refs:targetShape (Results 1 – 3 of 3) sorted by relevance
| /external/llvm-project/mlir/include/mlir/Dialect/Vector/ |
| D | VectorTransforms.h | 70 ArrayRef<int64_t> targetShape);
86 ArrayRef<int64_t> targetShape);
138 Optional<SmallVector<int64_t, 4>> targetShape = options.nativeShape(op); in matchAndRewrite() local
139 if (!targetShape) in matchAndRewrite()
141 auto maybeShapeRatio = shapeRatio(*maybeUnrollShape, *targetShape); in matchAndRewrite()
146 if (failed(unrollTransferWriteOp(rewriter, op, *targetShape))) in matchAndRewrite()
153 auto resultVector = unrollSingleResultVectorOp(rewriter, op, *targetShape); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Dialect/Vector/ |
| D | VectorTransforms.cpp | 235 ArrayRef<int64_t> targetShape, in initUnrolledVectorState() argument
240 getMappedElements(indexMap, targetShape, state.unrolledShape); in initUnrolledVectorState()
371 ArrayRef<int64_t> targetShape, in unrollSingleResultStructuredOp() argument
378 auto maybeUnrollFactors = shapeRatio(iterationBounds, targetShape); in unrollSingleResultStructuredOp()
390 targetShape, unrolledVectorState[i], builder); in unrollSingleResultStructuredOp()
409 computeElementOffsetsFromVectorSliceOffsets(targetShape, vectorOffsets); in unrollSingleResultStructuredOp()
458 vector::ContractionOp contractionOp, ArrayRef<int64_t> targetShape, in getVectorContractionOpUnrollState() argument
487 ArrayRef<int64_t> targetShape, in getVectorElementwiseOpUnrollState() argument
591 ArrayRef<int64_t> targetShape, in unrollTransferReadOp() argument
596 SmallVector<int64_t, 4> strides(targetShape.size(), 1); in unrollTransferReadOp()
[all …]
|
| /external/llvm-project/mlir/include/mlir/Interfaces/ |
| D | VectorInterfaces.td | 28 `targetShape`. Return `None` if the op cannot be unrolled to the target
|