diff options
| author | Frederick Liardet <frederick.liardet@arm.com> | 2022-08-24 10:07:25 +0100 |
|---|---|---|
| committer | Frederick Liardet <frederick.liardet@arm.com> | 2022-08-31 16:47:30 +0100 |
| commit | 62c1e95dc2f438e027f94e1dd2f486631489b189 (patch) | |
| tree | 4460d880bcd6bbfe155a279c869de659e0e22116 /python/tosa_serializer.py | |
| parent | 545a508429afe1d22760563d252839e13ecd12a3 (diff) | |
| download | serialization_lib-experimental/tf28_compat.tar.gz | |
Add tosa identification numbersexperimental/tf28_compat
Signed-off-by: Frederick Liardet <frederick.liardet@arm.com>
Change-Id: Idc5668fb01dc8dd657beb911d2cb9295f9275ac6
Diffstat (limited to 'python/tosa_serializer.py')
| -rw-r--r-- | python/tosa_serializer.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/python/tosa_serializer.py b/python/tosa_serializer.py index 6915c83..b11f9cd 100644 --- a/python/tosa_serializer.py +++ b/python/tosa_serializer.py @@ -60,6 +60,9 @@ DTypeNames = [ "FLOAT", ] +# File identifier needs to be kept in sync with schema/tosa.fbs +TOSA_GRAPH_IDENTIFIER = b"\x54\x4F\x53\x41" + ByteMask = np.uint64(0xFF) @@ -713,7 +716,7 @@ class TosaSerializer: TosaGraph.TosaGraphAddBlocks(builder, fbv_bb) graph = TosaGraph.TosaGraphEnd(builder) - self.builder.Finish(graph) + self.builder.Finish(graph, TOSA_GRAPH_IDENTIFIER) return self.builder.Output() def writeJson(self, tosa_filename): |