Home
last modified time | relevance | path

Searched refs:dimPair (Results 1 – 2 of 2) sorted by relevance

/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorOps.cpp276 for (auto &dimPair : map) { in verifyDimMap() local
277 if (dimPair.first < 0 || dimPair.first >= lhsType.getRank() || in verifyDimMap()
278 dimPair.second < 0 || dimPair.second >= rhsType.getRank() || in verifyDimMap()
279 lhsType.getDimSize(dimPair.first) != rhsType.getDimSize(dimPair.second)) in verifyDimMap()
292 for (auto &dimPair : contractingDimMap) { in verifyOutputShape() local
293 lhsContractingDimSet.insert(dimPair.first); in verifyOutputShape()
294 rhsContractingDimSet.insert(dimPair.second); in verifyOutputShape()
297 for (auto &dimPair : batchDimMap) in verifyOutputShape() local
298 rhsBatchDimSet.insert(dimPair.second); in verifyOutputShape()
DVectorTransforms.cpp1971 for (auto &dimPair : contractingDimMap) { in matchAndRewrite() local
1972 lhsContractingDimSet.insert(dimPair.first); in matchAndRewrite()
1973 rhsContractingDimSet.insert(dimPair.second); in matchAndRewrite()