Add lowering for aten.clone and aten.view_copy (#602)

Summary: Pull Request resolved: #602

Reviewed By: cccclai

Differential Revision: D49897893

Pulled By: digantdesai

fbshipit-source-id: d08d3e90e45b7741162ca083ca3da5e8ca4251ee

Author: Jerry-Ge

backends/arm/arm_backend.py

@@ -74,6 +74,8 @@ def is_node_supported(self, submodules, node: torch.fx.Node) -> bool:
             exir_ops.edge.aten._native_batch_norm_legit_no_training.default,
             exir_ops.edge.aten.avg_pool2d.default,
             exir_ops.edge.aten._softmax.default,
+            exir_ops.edge.aten.view_copy.default,
+            exir_ops.edge.aten.clone.default,
             operator.getitem,
             exir_ops.edge.quantized_decomposed.quantize_per_tensor.default,
             exir_ops.edge.quantized_decomposed.dequantize_per_tensor.default,
@@ -762,10 +764,18 @@ def preprocess(  # noqa: C901
                         [outp.name],
                         attr_mul,
                     )
+                elif exir_ops.edge.aten.view_copy.default == node.target:
+                    attr = ts.TosaSerializerAttribute()
+                    new_shape = inputs[1].special
+                    attr.ReshapeAttribute(new_shape)
+                    tosa_fb.addOperator(
+                        TosaOp.Op().RESHAPE, [inputs[0].name], [outp.name], attr
+                    )
                 elif node.target in [
                     operator.getitem,
                     tosa_quant_utils.q_op,
                     tosa_quant_utils.dq_op,
+                    exir_ops.edge.aten.clone.default,
                 ]:
                     item_name = inputs[0].name
                     ## Simply add an identityOp

backends/arm/test/test_models.py

@@ -34,6 +34,34 @@ class TorchBuilder:
     def __init__(self):
         pass
 
+    @register_test
+    class simple_clone(torch.nn.Module):
+        inputs = {
+            TosaProfile.BI: (torch.ones(10),),
+            TosaProfile.MI: (torch.ones(10),),
+        }
+
+        def __init__(self):
+            super().__init__()
+
+        def forward(self, x):
+            x = x.clone()
+            return x
+
+    @register_test
+    class simple_view(torch.nn.Module):
+        inputs = {
+            TosaProfile.BI: (torch.ones(10),),
+            TosaProfile.MI: (torch.ones(10),),
+        }
+
+        def __init__(self):
+            super().__init__()
+
+        def forward(self, x):
+            x = x.view(2, 5)
+            return x
+
     @register_test
     class simple_add(torch.nn.Module):
         inputs = {