Add tool to find out which ops are needed by a graph

mozilla · Aug 2, 2018 · d2be00f · d2be00f
1 parent 400c9c0
commit d2be00f
Showing 1 changed file with 11 additions and 0 deletions.
diff --git a/bin/ops_in_graph.py b/bin/ops_in_graph.py
@@ -0,0 +1,11 @@
+#!/usr/bin/env python
+# -*- coding: utf-8 -*-
+
+import tensorflow as tf
+import sys
+
+with tf.gfile.FastGFile(sys.argv[1], 'rb') as fin:
+    graph_def = tf.GraphDef()
+    graph_def.MergeFromString(fin.read())
+
+    print('\n'.join(sorted(set(n.op for n in graph_def.node))))