Home
last modified time | relevance | path

Searched refs:loopToVectorDim (Results 1 – 5 of 5) sorted by relevance

/llvm-project-15.0.7/mlir/lib/Dialect/Vector/Utils/
H A DVectorUtils.cpp200 const DenseMap<Operation *, unsigned> &loopToVectorDim) { in makePermutationMap() argument
204 auto it = loopToVectorDim.find(forInst); in makePermutationMap()
205 if (it != loopToVectorDim.end()) { in makePermutationMap()
214 const DenseMap<Operation *, unsigned> &loopToVectorDim) { in makePermutationMap() argument
215 return makePermutationMap(op->getBlock(), indices, loopToVectorDim); in makePermutationMap()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Vector/Utils/
H A DVectorUtils.h145 const DenseMap<Operation *, unsigned> &loopToVectorDim);
148 const DenseMap<Operation *, unsigned> &loopToVectorDim);
/llvm-project-15.0.7/mlir/lib/Dialect/Affine/Transforms/
H A DSuperVectorize.cpp631 strategy->loopToVectorDim[loop] = in vectorizeLoopIfProfitable()
1077 if (forOp && strategy->loopToVectorDim.count(forOp) == 0) in isUniformDefinition()
1080 for (auto loopToDim : strategy->loopToVectorDim) { in isUniformDefinition()
1272 auto loopToVecDimIt = strategy.loopToVectorDim.find(forOp); in vectorizeAffineForOp()
1273 bool isLoopVecDim = loopToVecDimIt != strategy.loopToVectorDim.end(); in vectorizeAffineForOp()
/llvm-project-15.0.7/mlir/include/mlir/Dialect/Affine/
H A DUtils.h93 DenseMap<Operation *, unsigned> loopToVectorDim; member
/llvm-project-15.0.7/mlir/test/lib/Dialect/Affine/
H A DTestVectorizationUtils.cpp236 strategy.loopToVectorDim[outermostLoop] = 0; in testVecAffineLoopNest()