Home
last modified time | relevance | path

Searched defs:targetShape (Results 1 – 2 of 2) sorted by relevance

/external/llvm-project/mlir/include/mlir/Dialect/Vector/
DVectorTransforms.h138 Optional<SmallVector<int64_t, 4>> targetShape = options.nativeShape(op); in matchAndRewrite() local
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorTransforms.cpp235 ArrayRef<int64_t> targetShape, in initUnrolledVectorState()
371 ArrayRef<int64_t> targetShape, in unrollSingleResultStructuredOp()
458 vector::ContractionOp contractionOp, ArrayRef<int64_t> targetShape, in getVectorContractionOpUnrollState()
487 ArrayRef<int64_t> targetShape, in getVectorElementwiseOpUnrollState()
591 ArrayRef<int64_t> targetShape, in unrollTransferReadOp()
635 ArrayRef<int64_t> targetShape) { in unrollTransferWriteOp()
666 ArrayRef<int64_t> targetShape) { in unrollSingleResultVectorOp()