Searched refs:sourceDim (Results 1 – 3 of 3) sorted by relevance
35 unsigned sourceDim = 0; in getReassociationIndicesForCollapse() local41 while (sourceDim < sourceShape.size()) { in getReassociationIndicesForCollapse()51 sourceDim < sourceShape.size()) { in getReassociationIndicesForCollapse()52 prodOfCollapsedDims *= sourceShape[sourceDim]; in getReassociationIndicesForCollapse()53 currIndices.push_back(sourceDim++); in getReassociationIndicesForCollapse()59 if (sourceShape[sourceDim] == ShapedType::kDynamicSize && in getReassociationIndicesForCollapse()67 sourceShape[sourceDim] != ShapedType::kDynamicSize) in getReassociationIndicesForCollapse()75 currIndices.push_back(sourceDim++); in getReassociationIndicesForCollapse()85 for (; sourceDim < sourceShape.size(); sourceDim++) { in getReassociationIndicesForCollapse()87 sourceShape[sourceDim] != 1) in getReassociationIndicesForCollapse()[all …]
168 SubscriptValue &sourceDim{sourceAt[dim - 1]}; in RTNAME() local169 sourceDim = dimLB + shiftCount % dimExtent; in RTNAME()170 if (sourceDim < dimLB) { in RTNAME()171 sourceDim += dimExtent; in RTNAME()175 if (++sourceDim == dimLB + dimExtent) { in RTNAME()176 sourceDim = dimLB; in RTNAME()189 SubscriptValue extent{sourceDim.Extent()}; in RTNAME()191 SubscriptValue lb{sourceDim.LowerBound()}; in RTNAME()266 sourceDim = dimLB + shiftCount; in RTNAME()268 if (sourceDim >= dimLB && sourceDim < dimLB + dimExtent) { in RTNAME()[all …]
102 Value sourceDim = builder.create<tensor::DimOp>(loc, src, sourceDimPos); in getExpandedOutputDimFromInputShape() local107 sourceDim)[0]; in getExpandedOutputDimFromInputShape()