Lines Matching defs:sourceDim

36   unsigned sourceDim = 0;
42   while (sourceDim < sourceShape.size()) {
50     while (sourceDim < (sourceShape.size() - 1) &&
51            sourceShape[sourceDim] != ShapedType::kDynamic &&
52            prodOfCollapsedDims * sourceShape[sourceDim] < currTargetShape) {
53       prodOfCollapsedDims *= sourceShape[sourceDim];
54       currIndices.push_back(sourceDim++);
60     if (sourceShape[sourceDim] == ShapedType::kDynamic &&
67         sourceShape[sourceDim] != ShapedType::kDynamic)
72     if (prodOfCollapsedDims * sourceShape[sourceDim] != currTargetShape)
75     currIndices.push_back(sourceDim++);
85   for (; sourceDim < sourceShape.size(); sourceDim++) {
86     if (sourceShape[sourceDim] != ShapedType::kDynamic &&
87         sourceShape[sourceDim] != 1)
91       reassociationMap.back().push_back(sourceDim);