"git@developer.sourcefind.cn:OpenDAS/torch-scatter.git" did not exist on "f32831642a2e8f001c04acad10564e342211eeb1"
Commit d32d957a authored by Gonzalo Gasca Meza's avatar Gonzalo Gasca Meza Committed by Karmel Allison
Browse files

Update tensorrt.py (#6034)

Changes in Tensorflow contrib:
https://github.com/tensorflow/tensorflow/commit/304faf0444260912b6996d39227417c09561c37e

Causes tensortt.py to fail. 

```
AttributeError: 'module' object has no attribute 'get_signature_def_by_key'
```
Replace `tf.contrib.saved_model.get_signature_def_by_key` with `meta_graph_def.signature_def`
parent 0e05682d
...@@ -182,10 +182,9 @@ def convert_savedmodel_to_frozen_graph(savedmodel_dir, output_dir): ...@@ -182,10 +182,9 @@ def convert_savedmodel_to_frozen_graph(savedmodel_dir, output_dir):
Returns: Returns:
Frozen Graph definition for use. Frozen Graph definition for use.
""" """
meta_graph = get_serving_meta_graph_def(savedmodel_dir) meta_graph_def = get_serving_meta_graph_def(savedmodel_dir)
signature_def = tf.contrib.saved_model.get_signature_def_by_key( signature_def = meta_graph_def.signature_def[
meta_graph, tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY]
tf.saved_model.signature_constants.DEFAULT_SERVING_SIGNATURE_DEF_KEY)
outputs = [v.name for v in signature_def.outputs.itervalues()] outputs = [v.name for v in signature_def.outputs.itervalues()]
output_names = [node.split(":")[0] for node in outputs] output_names = [node.split(":")[0] for node in outputs]
...@@ -193,7 +192,7 @@ def convert_savedmodel_to_frozen_graph(savedmodel_dir, output_dir): ...@@ -193,7 +192,7 @@ def convert_savedmodel_to_frozen_graph(savedmodel_dir, output_dir):
graph = tf.Graph() graph = tf.Graph()
with tf.Session(graph=graph) as sess: with tf.Session(graph=graph) as sess:
tf.saved_model.loader.load( tf.saved_model.loader.load(
sess, meta_graph.meta_info_def.tags, savedmodel_dir) sess, meta_graph_def.meta_info_def.tags, savedmodel_dir)
frozen_graph_def = tf.graph_util.convert_variables_to_constants( frozen_graph_def = tf.graph_util.convert_variables_to_constants(
sess, graph.as_graph_def(), output_names) sess, graph.as_graph_def(), output_names)
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment