Compute Library
 21.08
tf_frozen_model_extractor.py File Reference

Go to the source code of this file.

Namespaces

 tf_frozen_model_extractor
 

Variables

list strings_to_remove = ["read", "/:0"]
 
dictionary permutations = { 1 : [0], 2 : [1, 0], 3 : [2, 1, 0], 4 : [3, 2, 0, 1]}
 
 parser = argparse.ArgumentParser('Extract TensorFlow net parameters')
 
 dest
 
 type
 
 str
 
 required
 
 True
 
 help
 
 False
 
 default
 
 action
 
 storeRes
 
 args = parser.parse_args()
 
 graph_def = tf.GraphDef()
 
 input_map
 
 None
 
 return_elements
 
 name
 
 op_dict
 
 producer_op_list
 
 varname = op_val.name
 
 t = op_val.eval()
 
 tT = t.transpose(permutations[len(t.shape)])