We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
There was an error while loading. Please reload this page.
1 parent c8b69c9 commit a1bb750Copy full SHA for a1bb750
mlir/lib/Dialect/Vector/IR/VectorOps.cpp
@@ -882,9 +882,8 @@ static LogicalResult verifyOutputShape(
882
lhsContractingDimSet.insert(dimPair.first);
883
rhsContractingDimSet.insert(dimPair.second);
884
}
885
- DenseSet<int64_t> rhsBatchDimSet;
886
- for (auto &dimPair : batchDimMap)
887
- rhsBatchDimSet.insert(dimPair.second);
+ DenseSet<int64_t> rhsBatchDimSet(llvm::from_range,
+ llvm::make_second_range(batchDimMap));
888
889
// Add free and batch dimensions from 'lhsType' to 'expectedResultDims'.
890
SmallVector<int64_t, 4> expectedResultDims;
0 commit comments