Searched refs:nextDim (Results 1 – 3 of 3) sorted by relevance
358 const int64_t nextDim = innerDim - 1; in getContiguousInnerDim() local359 if (shape[nextDim] == ShapedType::kDynamicSize) in getContiguousInnerDim()361 if (strides[nextDim] != innerSize) in getContiguousInnerDim()363 innerSize *= shape[nextDim]; in getContiguousInnerDim()364 innerDim = nextDim; in getContiguousInnerDim()
949 unsigned nextDim = 0; in canonicalizePromotedSymbols() local960 dimRemapping[i] = getAffineDimExpr(nextDim++, context); in canonicalizePromotedSymbols()970 *mapOrSet = mapOrSet->replaceDimsAndSymbols(dimRemapping, {}, nextDim, in canonicalizePromotedSymbols()1009 unsigned nextDim = 0; in canonicalizeMapOrSetAndOperands() local1015 dimRemapping[i] = getAffineDimExpr(nextDim++, context); in canonicalizeMapOrSetAndOperands()1051 nextDim, nextSym); in canonicalizeMapOrSetAndOperands()
1624 int64_t nextDim = 0; in verifyCollapsedShape() local1631 if (expandedDim != nextDim++) in verifyCollapsedShape()1673 } else if (nextDim != static_cast<int64_t>(expandedShape.size())) { in verifyCollapsedShape()1678 << ") inconsistent with number of reassociation indices (" << nextDim in verifyCollapsedShape()