Home
last modified time | relevance | path

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

/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
H A DVectorization.cpp310 initState(RewriterBase & rewriter,LinalgOp linalgOp,ArrayRef<int64_t> inputVectorSizes,ArrayRef<bool> inputScalableVecDims) initState() argument
1422 createWriteOrMaskedWrite(OpBuilder & builder,Location loc,Value input,SmallVector<OpFoldResult> destSizes,ArrayRef<int64_t> inputVectorSizes,bool useInBoundsInsteadOfMasking) createWriteOrMaskedWrite() argument
1493 vectorizeAsTensorPackOp(RewriterBase & rewriter,tensor::PackOp packOp,ArrayRef<int64_t> inputVectorSizes,SmallVectorImpl<Value> & newResults) vectorizeAsTensorPackOp() argument
1568 vectorizeAsTensorUnpackOp(RewriterBase & rewriter,tensor::UnPackOp unpackOp,ArrayRef<int64_t> inputVectorSizes,SmallVectorImpl<Value> & newResults) vectorizeAsTensorUnpackOp() argument
1696 vectorizeAsTensorPadOp(RewriterBase & rewriter,tensor::PadOp padOp,ArrayRef<int64_t> inputVectorSizes,SmallVectorImpl<Value> & newResults) vectorizeAsTensorPadOp() argument
1790 vectorizeUnPackOpPrecondition(tensor::UnPackOp unpackOp,ArrayRef<int64_t> inputVectorSizes) vectorizeUnPackOpPrecondition() argument
1810 vectorizeLinalgOpPrecondition(LinalgOp linalgOp,ArrayRef<int64_t> inputVectorSizes,bool vectorizeNDExtract,bool flatten1DDepthwiseConv) vectorizeLinalgOpPrecondition() argument
1878 vectorizePackOpPrecondition(tensor::PackOp packOp,ArrayRef<int64_t> inputVectorSizes) vectorizePackOpPrecondition() argument
1911 vectorizePadOpPrecondition(tensor::PadOp padOp,ArrayRef<int64_t> inputVectorSizes) vectorizePadOpPrecondition() argument
1937 vectorizeScalableVectorPrecondition(Operation * op,ArrayRef<int64_t> inputVectorSizes,ArrayRef<bool> inputScalableVecDims) vectorizeScalableVectorPrecondition() argument
1957 vectorizeOpPrecondition(Operation * op,ArrayRef<int64_t> inputVectorSizes,ArrayRef<bool> inputScalableVecDims,bool vectorizeNDExtract,bool flatten1DDepthwiseConv) vectorizeOpPrecondition() argument
2004 vectorize(RewriterBase & rewriter,Operation * op,ArrayRef<int64_t> inputVectorSizes,ArrayRef<bool> inputScalableVecDims,bool vectorizeNDExtract,bool flatten1DDepthwiseConv) vectorize() argument
[all...]
/llvm-project/mlir/lib/Dialect/Vector/Utils/
H A DVectorUtils.cpp373 isValidMaskedInputVector(ArrayRef<int64_t> shape,ArrayRef<int64_t> inputVectorSizes) isValidMaskedInputVector() argument