Move prim::layout for lite jit (pytorch#53781)

Summary:
Pull Request resolved: pytorch#53781

needed for running noise suppression model in lite interpreter

Test Plan: run model

Reviewed By: linbinyu

Differential Revision: D26967227

fbshipit-source-id: 19677fc796f1fb4423ebb11b5ffd9df5870a39cf

Author: jiatongzhou

torch/csrc/jit/runtime/register_prim_ops.cpp

@@ -106,6 +106,14 @@ RegisterOperators reg(
            push(stack, a.cpu());
          },
          aliasAnalysisFromSchema()),
+     Operator(
+         "prim::layout(Tensor a) -> int",
+         [](Stack* stack) {
+           at::Tensor a;
+           pop(stack, a);
+           push(stack, a.layout());
+         },
+         aliasAnalysisFromSchema()),
      Operator(
          prim::tolist,
          // This operator has to be unschematized because the return type

torch/csrc/jit/runtime/register_prim_ops_fulljit.cpp

@@ -333,14 +333,6 @@ RegisterOperators reg(
            }
          },
          aliasAnalysisFromSchema()),
-     Operator(
-         "prim::layout(Tensor a) -> int",
-         [](Stack* stack) {
-           at::Tensor a;
-           pop(stack, a);
-           push(stack, a.layout());
-         },
-         aliasAnalysisFromSchema()),
      Operator(
          "prim::index(Device self) -> int?",
          [](Stack* stack) {