Deserialize to named data store output (pytorch#15484)

This PR was created by the merge bot to help merge the original PR into
the main branch.
ghstack PR number: pytorch#15469 by
@lucylq
^ Please use this as the source of truth for the PR details, comments,
and reviews
ghstack PR base:
https://github.com/pytorch/executorch/tree/gh/lucylq/122/base
ghstack PR head:
https://github.com/pytorch/executorch/tree/gh/lucylq/122/head
Merge bot PR base: https://github.com/pytorch/executorch/tree/main
Merge bot PR head:
https://github.com/pytorch/executorch/tree/gh/lucylq/122/orig
Differential Revision:
[D83510300](https://our.internmc.facebook.com/intern/diff/D83510300/)
@diff-train-skip-merge

Co-authored-by: lucylq <[email protected]>

Author: 2 people

extension/flat_tensor/serialize/TARGETS

@@ -29,6 +29,7 @@ runtime.python_library(
     ],
     visibility = [
         "//executorch/...",
+        "@EXECUTORCH_CLIENTS",
     ],
     deps = [
         ":schema",

extension/flat_tensor/serialize/serialize.py

@@ -20,6 +20,7 @@
 from executorch.exir._serialize._cord import Cord
 from executorch.exir._serialize._dataclass import _DataclassEncoder, _json_to_dataclass
 from executorch.exir._serialize._flatbuffer import _flatc_compile, _flatc_decompile
+from executorch.exir._serialize._named_data_store import NamedDataStoreOutput
 from executorch.exir._serialize._program import _insert_flatbuffer_header
 from executorch.exir._serialize.data_serializer import (
     DataEntry,
@@ -389,6 +390,8 @@ def serialize(
     def deserialize(self, blob: Cord) -> DataPayload:
         """
         Deserializes a flat_tensor blob into a list of tensor metadata and tensors.
+
+        Note: deserialization does not preserve alignment information.
         """
 
         data = bytes(blob)
@@ -436,3 +439,14 @@ def deserialize(self, blob: Cord) -> DataPayload:
             payload.named_data[named_data.key] = entry
 
         return payload
+
+    def deserialize_to_named_data_store_output(
+        self, blob: bytes, name: str
+    ) -> NamedDataStoreOutput:
+        bytes = Cord(blob)
+        data_payload = self.deserialize(bytes)
+        return NamedDataStoreOutput(
+            buffers=data_payload.buffers,
+            pte_data={},
+            external_data={name: data_payload.named_data},
+        )