Lines Matching defs:reshape

883 /// Fold dim of a tensor reshape operation to a extract into the reshape's shape
890     auto reshape = dim.getSource().getDefiningOp<ReshapeOp>();
892     if (!reshape)
900         rewriter.create<ExtractOp>(loc, reshape.getShape(), dim.getIndex());
1122 /// Try to remove a tensor operation if it would only reshape a constant.
1601   setNameFn(getResult(), "reshape");
1633                          "reshape to statically-ranked tensor type");
1647   // If the producer of operand 'source' is another 'tensor.reshape' op, use the
1648   // producer's input instead as the original tensor to reshape. This could
1662   // reshape has no effect, even if the tensor is dynamically shaped.
1700 // Reassociative reshape ops