mirrors/DeepSpeech
2
0
Fork 0
mirror of https://github.com/mozilla/DeepSpeech.git synced 2025-10-26 11:19:39 +00:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Reuben Morais 2018-03-20 20:37:29 -03:00
parent 400c9c0e5d
commit d2be00fad0
1 changed files with 11 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

11
bin/ops_in_graph.py Executable file
View File

@ -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))))