We read every piece of feedback, and take your input very seriously.
To see all available qualifiers, see our documentation.
1 parent bfa1a60 commit 2021be9Copy full SHA for 2021be9
Athos/TFCompiler/DumpTFMtData.py
@@ -52,6 +52,8 @@ def save_graph_metadata(output_tensor, sess, feed_dict):
52
optimized_graph_def = TransformGraph(graph_def, [], [output_tensor.name], transforms)
53
with open('./graphDef.mtdata', 'w') as f:
54
f.write(str(optimized_graph_def))
55
+ with open('./graphDef.bin', 'w') as f:
56
+ f.write(optimized_graph_def.SerializeToString())
57
58
# Save size information for tensors on which output depends
59
tensors_to_evaluate = []
0 commit comments