Dump Tensorflow graphs in binary.

mkskeller · May 25, 2020 · 2021be9 · 2021be9
1 parent bfa1a60
commit 2021be9
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/Athos/TFCompiler/DumpTFMtData.py b/Athos/TFCompiler/DumpTFMtData.py
@@ -52,6 +52,8 @@ def save_graph_metadata(output_tensor, sess, feed_dict):
   optimized_graph_def = TransformGraph(graph_def, [], [output_tensor.name], transforms)
   with open('./graphDef.mtdata', 'w') as f:
     f.write(str(optimized_graph_def))
+  with open('./graphDef.bin', 'w') as f:
+    f.write(optimized_graph_def.SerializeToString())
 
   # Save size information for tensors on which output depends
   tensors_to_evaluate = []