Lines Matching defs:dimIdx

484   for (auto [dimIdx, dimSize] :
487                       [dimIdx = dimIdx](int64_t reductionDimIdx) {
488                         return reductionDimIdx == static_cast<int64_t>(dimIdx);
491       scalableDims.push_back(sourceScalableDims[dimIdx]);
1056   for (auto [dimIdx, dimSize] : llvm::enumerate(lhsType.getShape())) {
1057     maskShape[lhsIdxMap.getDimPosition(dimIdx)] = dimSize;
1058     maskShapeScalableDims[lhsIdxMap.getDimPosition(dimIdx)] =
1059         lhsType.getScalableDims()[dimIdx];
1061   for (auto [dimIdx, dimSize] : llvm::enumerate(rhsType.getShape())) {
1062     maskShape[rhsIdxMap.getDimPosition(dimIdx)] = dimSize;
1063     maskShapeScalableDims[rhsIdxMap.getDimPosition(dimIdx)] =
1064         rhsType.getScalableDims()[dimIdx];
2168     for (size_t dimIdx = 0; !allFalse && dimIdx < extractOpPos.size();
2169          dimIdx++) {
2170       int64_t pos = extractOpPos[dimIdx];
2171       Value operand = maskOperands[dimIdx];
2186       } else if (createMaskBound < maskType.getDimSize(dimIdx)) {
2501   for (int64_t dimIdx = 0; dimIdx < srcRank; ++dimIdx) {
2507     int64_t srcDim = srcVectorType.getDimSize(dimIdx);
2508     int64_t dstDim = dstVectorType.getDimSize(lead + dimIdx);
2513     bool srcDimScalableFlag = srcVectorType.getScalableDims()[dimIdx];
2514     bool dstDimScalableFlag = dstVectorType.getScalableDims()[lead + dimIdx];