From dde29e25ed00cbaba8ad21db4e7dc536cc9f3580 Mon Sep 17 00:00:00 2001 From: wuchenghui Date: Thu, 8 Mar 2018 17:09:12 +0800 Subject: [PATCH] CPU+GPU+DSP full throughput benchmark --- python/tools/tf_converter_lib.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/python/tools/tf_converter_lib.py b/python/tools/tf_converter_lib.py index 5d1ea941..cbedb094 100644 --- a/python/tools/tf_converter_lib.py +++ b/python/tools/tf_converter_lib.py @@ -344,7 +344,7 @@ class TFConverter(object): final_op = op self.resolved_ops[op.name] = 1 - if len(self.tf_graph[op.name]) == 1 and self.tf_graph[op.name][0].type == 'BiasAdd' : + if len(self.tf_graph.get(op.name, [])) == 1 and self.tf_graph[op.name][0].type == 'BiasAdd': bias_add_op = self.tf_graph[op.name][0] if self.device == 'gpu': output_name = self.add_buffer_to_image(get_input_tensor(bias_add_op, 1).name, "ARGUMENT") -- GitLab