Lines Matching defs:outerDimsPerm
121 // outerDimsPerm = [1, 2, 0]
177 SmallVector<int64_t> outerDimsPerm;
190 outerDimsPerm.push_back(currentPositionTileLoops.lookup(loopIdx));
192 return outerDimsPerm;
262 SmallVector<int64_t> outerDimsPerm;
264 outerDimsPerm = computeOuterDims(packInfo.outerDimsOnDomainPerm, exprs);
279 // transposition on the pack using outerDimsPerm.
280 if (!outerDimsPerm.empty()) {
282 for (const auto &en : enumerate(outerDimsPerm))
290 if (innerDimsPos.empty() && outerDimsPerm.empty())
294 b, loc, opOperand->get(), innerTileSizes, innerDimsPos, outerDimsPerm);
297 /*padding=*/std::nullopt, outerDimsPerm);
523 ArrayRef<int64_t> outerDimsPerm = packOp.getOuterDimsPerm();
527 outerDimsPerm);
530 /*padding=*/std::nullopt, outerDimsPerm);
535 if (!outerDimsPerm.empty()) {
536 applyPermutationToVector<OpFoldResult>(lowPad, outerDimsPerm);
537 applyPermutationToVector<OpFoldResult>(highPad, outerDimsPerm);
553 rewriter, loc, newPadOp, mixedTiles, innerDimsPos, outerDimsPerm);
555 loc, newPadOp, unpackEmpty, innerDimsPos, mixedTiles, outerDimsPerm);
656 ArrayRef<int64_t> outerDimsPerm = packOp.getOuterDimsPerm();
679 for (auto outerPos : outerDimsPerm) {
697 applyPermutationAndReindexReassoc(newReassocIndices, outerDimsPerm);
762 ArrayRef<int64_t> outerDimsPerm = packOp.getOuterDimsPerm();
763 if (!outerDimsPerm.empty() && !isIdentityPermutation(outerDimsPerm)) {
813 projectedInnerDimsPos, /*outerDimsPerm=*/SmallVector<int64_t>{});
822 projectedInnerDimsPos, /*outerDimsPerm=*/SmallVector<int64_t>{});
826 /*outerDimsPerm=*/SmallVector<int64_t>{});
902 ArrayRef<int64_t> outerDimsPerm = unPackOp.getOuterDimsPerm();
928 for (auto outerPos : outerDimsPerm) {
939 applyPermutationAndReindexReassoc(newReassocIndices, outerDimsPerm);
1106 auto outerDimsPerm = destPack.getOuterDimsPerm();
1113 mixedTiles, outerDimsPerm)
1172 ArrayRef<int64_t> outerDimsPerm = unpackOp.getOuterDimsPerm();
1175 if (!outerDimsPerm.empty()) {
1176 applyPermutationToVector<OpFoldResult>(lowPad, outerDimsPerm);
1177 applyPermutationToVector<OpFoldResult>(highPad, outerDimsPerm);
1195 unpackOp.getMixedTiles(), outerDimsPerm);