DeepSpeech/bin/ops_in_graph.py

12 строки
271 B
Python
Executable File

#!/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.ParseFromString(fin.read())
print('\n'.join(sorted(set(n.op for n in graph_def.node))))