Lines Matching refs:sourceShape
31 mlir::getReassociationIndicesForCollapse(ArrayRef<int64_t> sourceShape, in getReassociationIndicesForCollapse() argument
33 if (sourceShape.size() <= targetShape.size()) in getReassociationIndicesForCollapse()
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()
59 if (sourceShape[sourceDim] == ShapedType::kDynamicSize && in getReassociationIndicesForCollapse()
67 sourceShape[sourceDim] != ShapedType::kDynamicSize) in getReassociationIndicesForCollapse()
72 if (prodOfCollapsedDims * sourceShape[sourceDim] != currTargetShape) in getReassociationIndicesForCollapse()
85 for (; sourceDim < sourceShape.size(); sourceDim++) { in getReassociationIndicesForCollapse()
86 if (sourceShape[sourceDim] != ShapedType::kDynamicSize && in getReassociationIndicesForCollapse()
87 sourceShape[sourceDim] != 1) in getReassociationIndicesForCollapse()