Lines Matching defs:sourceShape

32 mlir::getReassociationIndicesForCollapse(ArrayRef<int64_t> sourceShape,
34   if (sourceShape.size() <= targetShape.size())
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];
60     if (sourceShape[sourceDim] == ShapedType::kDynamic &&
67         sourceShape[sourceDim] != ShapedType::kDynamic)
72     if (prodOfCollapsedDims * sourceShape[sourceDim] != currTargetShape)
85   for (; sourceDim < sourceShape.size(); sourceDim++) {
86     if (sourceShape[sourceDim] != ShapedType::kDynamic &&
87         sourceShape[sourceDim] != 1)