[executorch][weight sharing] Introduce NamedData to PTE schema

Pull Request resolved: #8695

See 'Schema Changes' in the [RFC](
ghstack-source-id: 268535712
@exported-using-ghexport

Differential Revision: [D69430152](https://our.internmc.facebook.com/intern/diff/D69430152/)

Author: lucylq

exir/schema.py

@@ -290,6 +290,12 @@ class SubsegmentOffsets:
     offsets: List[int]
 
 
+@dataclass
+class NamedData:
+    key: str
+    segment_index: int
+
+
 @dataclass
 class Program:
     version: int
@@ -299,3 +305,4 @@ class Program:
     segments: List[DataSegment]
     constant_segment: SubsegmentOffsets
     mutable_data_segments: Optional[List[SubsegmentOffsets]] = None
+    named_data: Optional[List[NamedData]] = None

schema/program.fbs

@@ -431,6 +431,17 @@ table SubsegmentOffsets {
   offsets: [uint64];
 }
 
+// Attributes a name to data referenced by Program.segments. Used when data is
+// referenced by multiple users, in cases where indices are not guaranteed to
+// be consistent across the users.
+table NamedData {
+  // The unique id of the data blob.
+  key: string;
+
+  // Index of the segment in Program.segments.
+  segment_index: uint32;
+}
+
 table Program {
   // Schema version.
   version: uint;
@@ -468,6 +479,11 @@ table Program {
   // constant memory, copying it over, and then being unable to release the
   // constant segment. No two elements should point to the same segment.
   mutable_data_segments: [SubsegmentOffsets];
+
+  // [Optional] List of blobs keyed by a unique name. Note that multiple
+  // 'NamedData' entries could point to the same segment index. Stored in
+  // segments attached to the PTE file.
+  named_data: [NamedData];
 }
 
 root_type Program;