Lines Matching refs:sourceDim
35 unsigned sourceDim = 0; in getReassociationIndicesForCollapse() local
41 while (sourceDim < sourceShape.size()) { in getReassociationIndicesForCollapse()
49 while (sourceShape[sourceDim] != ShapedType::kDynamicSize && in getReassociationIndicesForCollapse()
50 prodOfCollapsedDims * sourceShape[sourceDim] < currTargetShape && 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()
72 if (prodOfCollapsedDims * sourceShape[sourceDim] != currTargetShape) in getReassociationIndicesForCollapse()
75 currIndices.push_back(sourceDim++); in getReassociationIndicesForCollapse()
85 for (; sourceDim < sourceShape.size(); sourceDim++) { in getReassociationIndicesForCollapse()
86 if (sourceShape[sourceDim] != ShapedType::kDynamicSize && in getReassociationIndicesForCollapse()
87 sourceShape[sourceDim] != 1) in getReassociationIndicesForCollapse()
91 reassociationMap.back().push_back(sourceDim); in getReassociationIndicesForCollapse()