Move opinput api.py argtype.py and dtypes.py to edge dialect (#249)

Summary:
Pull Request resolved: #249

Move argument, dtype constraints runner logic from `kernels/test/fb/opinput` to `exir/dialects/edge/*`. Also did a restructure of `exir/dialects/edge`:

```
fbcode/executorch/exir/dialects/edge/
├── arg                    # Edge op argument model and types
│   ├── model.py
│   ├── TARGETS
│   └── type.py
├── BUCK
├── dtype                  # Edge supported dtype and logic to extract dtype constraints from core ATen ops.
│   ├── runner.py
│   ├── supported.py
│   └── TARGETS
├── edge.yaml
├── op                     # Torch operator getter and other useful APIs
│   ├── api.py
│   └── TARGETS
├── _ops.py
├── spec                   # Edge dialect spec generator
│   ├── gen.py
│   ├── TARGETS
│   └── utils.py
├── TARGETS
└── test
    ├── BUCK
    ├── TARGETS
    ├── test_edge_ops.py
    └── test_edge_yaml.py
```

Notice that currently the moved code is not being used by edge dialect spec generator. Next diffs will handle that.

Reviewed By: manuelcandales

Differential Revision: D49065273

fbshipit-source-id: d807d8ff58c39695fe4a85ebfd3ea801e073d3a8

Author: larryliu0820

exir/dialects/edge/TARGETS

@@ -1,40 +1,8 @@
 load("@fbcode_macros//build_defs:export_files.bzl", "export_file")
-load("@fbcode_macros//build_defs:python_binary.bzl", "python_binary")
 load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
 
 oncall("executorch")
 
-python_library(
-    name = "yaml_generator_lib",
-    srcs = [
-        "yaml_generator.py",
-    ],
-    deps = [
-        "fbsource//third-party/pypi/expecttest:expecttest",  # @manual
-        "fbsource//third-party/pypi/ruamel-yaml:ruamel-yaml",
-        ":support_dtypes",
-        ":utils",
-        "//caffe2:torch",
-        "//pye/lib:eager_model_base",
-    ],
-)
-
-python_binary(
-    name = "yaml_generator",
-    srcs = [
-        "yaml_generator.py",
-    ],
-    main_module = "executorch.exir.dialects.edge.yaml_generator",
-    deps = [
-        "fbsource//third-party/pypi/expecttest:expecttest",  # @manual
-        "fbsource//third-party/pypi/ruamel-yaml:ruamel-yaml",
-        ":support_dtypes",
-        ":utils",
-        "//caffe2:torch",
-        "//pye/lib:eager_model_base",
-    ],
-)
-
 python_library(
     name = "lib",
     srcs = [
@@ -46,36 +14,16 @@ python_library(
     deps = [
         "fbsource//third-party/pypi/ruamel-yaml:ruamel-yaml",
         "fbsource//third-party/pypi/setuptools:setuptools",
-        ":support_dtypes",
-        ":utils",
         "//caffe2:torch",
         "//caffe2/torchgen:torchgen",
+        "//executorch/exir/dialects/edge/dtype:lib",
+        "//executorch/exir/dialects/edge/spec:lib",
         "//executorch/exir/operator:convert",
     ],
 )
 
-python_library(
-    name = "support_dtypes",
-    srcs = [
-        "support_dtypes.py",
-    ],
-    deps = [
-        "//caffe2:torch",
-    ],
-)
-
 export_file(
     name = "edge_yaml",
     src = "edge.yaml",
     visibility = ["//executorch/..."],
 )
-
-python_library(
-    name = "utils",
-    srcs = [
-        "utils.py",
-    ],
-    deps = [
-        "//caffe2:torch",
-    ],
-)

exir/dialects/edge/_ops.py

@@ -12,8 +12,8 @@
 
 import torch
 
-from executorch.exir.dialects.edge.support_dtypes import regular_tensor_str_to_dtypes
-from executorch.exir.dialects.edge.utils import (
+from executorch.exir.dialects.edge.dtype.supported import regular_tensor_str_to_dtypes
+from executorch.exir.dialects.edge.spec.utils import (
     get_tensor_variable_names,
     get_torch_op_overload,
 )

exir/dialects/edge/arg/TARGETS

@@ -0,0 +1,14 @@
+load("@fbcode_macros//build_defs:python_library.bzl", "python_library")
+
+oncall("ai_infra_mobile_platform")
+
+python_library(
+    name = "lib",
+    srcs = [
+        "model.py",
+        "type.py",
+    ],
+    deps = [
+        "//caffe2:torch",
+    ],
+)