Update serializer to sync with export serializer (#4264)

Summary:
Pull Request resolved: #4264

This diff brings in the latest export serializer changes from `torch/_export/serde` and then fixes the exir serializer to use these changes. This is a temporary workaround until the serialization extensiblity that zhxchen17 is working on is completed. Then we'll not have to copy over changes from export serializer manually, instead we'll just extend it to handle things like delegate calls and other edge dialect specific things.

For now we need this to unblock ASR use case and hence i'm manually syncing and updating for now.

Reviewed By: JacobSzwejbka

Differential Revision: D57071033

fbshipit-source-id: 4408e2a0740b661e3a3555f800d1567ef10d4ea8

Author: tarun292

.ci/docker/ci_commit_pins/pytorch.txt

@@ -1 +1 @@
-c017c97333dfb9d17f2e5357980241827e50e8d5
+4e39cdceb1414b2d416339866a5bb044fbed4977

.lintrunner.toml

@@ -7,6 +7,7 @@ exclude_patterns = [
     'third-party/**',
     '**/third-party/**',
     '.github/scripts/**',
+    'exir/serde/**',
 ]
 command = [
     'python',
@@ -37,6 +38,7 @@ include_patterns = [
 exclude_patterns = [
     'third-party/**',
     '**/third-party/**',
+    'exir/serde/**',
 ]
 command = [
     'python',

exir/serde/TARGETS

@@ -8,11 +8,14 @@ python_library(
     name = "serialize",
     srcs = [
         "export_serialize.py",
+        "schema.py",
+        "schema_check.py",
         "serialize.py",
+        "union.py",
+        "upgrade.py",
     ],
     deps = [
         "fbsource//third-party/pypi/sympy:sympy",
-        ":schema",
         "//caffe2:torch",
         "//executorch/exir:delegate",
         "//executorch/exir:lowered_backend_module",
@@ -23,13 +26,3 @@ python_library(
         "//executorch/exir/dialects/edge:lib",
     ],
 )
-
-python_library(
-    name = "schema",
-    srcs = [
-        "schema.py",
-    ],
-    deps = [
-        "//caffe2:torch",
-    ],
-)