Add sample input for pixel_shuffle (#364)

Summary: Pull Request resolved: #364

Reviewed By: digantdesai

Differential Revision: D49290179

fbshipit-source-id: cb651a0ff37e09f68c64c620e41d6741f07fbcff

Author: manuelcandales

exir/dialects/edge/op/sample_input.py

@@ -930,6 +930,15 @@
             Return(ArgType.Tensor),
         ],
     },
+    "pixel_shuffle.default": {  # (Tensor self, int upscale_factor) -> Tensor
+        "args": [
+            InArg(ArgType.Tensor, size=[2, 4, 1, 3]),
+            InArg(ArgType.Param, value=2),
+        ],
+        "returns": [
+            Return(ArgType.Tensor, size=[2, 1, 2, 6]),
+        ],
+    },
     "pow.Tensor_Scalar": {  # (Tensor self, Scalar exponent) -> Tensor
         "args": [
             InArg(ArgType.Tensor),