From 4aa7ecc78b73f95d07e3ae7726c0ec754ef0a9d5 Mon Sep 17 00:00:00 2001 From: 692307 Date: Thu, 27 Jun 2019 09:16:43 +0800 Subject: [PATCH] add safe_remove_nodes function in gemm_rewriter.py --- tf2onnx/rewriter/gemm_rewriter.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tf2onnx/rewriter/gemm_rewriter.py b/tf2onnx/rewriter/gemm_rewriter.py index 37a07999b..1bb96ce1c 100644 --- a/tf2onnx/rewriter/gemm_rewriter.py +++ b/tf2onnx/rewriter/gemm_rewriter.py @@ -118,7 +118,7 @@ def rewrite_gemm(g, ops): ops.append(gemm) g.replace_all_inputs(ops, add_node.output[0], gemm.output[0]) + to_delete = [add_node, matmul_node] + g.safe_remove_nodes(to_delete) - # to_delete = [add_node, matmul_node, input_c_node] - # g.delete_nodes_without_dependency(to_delete) return ops