From 8b5a7fb29e9a28e360090ade7872b329cba226b4 Mon Sep 17 00:00:00 2001 From: Gwena - Workstation Date: Fri, 11 Feb 2022 16:29:38 +0900 Subject: [PATCH] Fix concat issue by enabling access to the 'optimizers' variable 'optimizers' variable is in 'optimizer.optimize_graph' in '_convert_common' Signed-off-by: Gwena - Workstation --- tf2onnx/convert.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/tf2onnx/convert.py b/tf2onnx/convert.py index d2f65f812..24c185e0a 100644 --- a/tf2onnx/convert.py +++ b/tf2onnx/convert.py @@ -138,7 +138,7 @@ def default_custom_op_handler(ctx, node, name, args): def _convert_common(frozen_graph, name="unknown", large_model=False, output_path=None, - output_frozen_graph=None, custom_ops=None, custom_op_handlers=None, **kwargs): + output_frozen_graph=None, custom_ops=None, custom_op_handlers=None, optimizers=None, **kwargs): """Common processing for conversion.""" model_proto = None @@ -165,7 +165,7 @@ def _convert_common(frozen_graph, name="unknown", large_model=False, output_path catch_errors = constants.ENV_TF2ONNX_CATCH_ERRORS.upper() == "TRUE" else: catch_errors = not large_model - onnx_graph = optimizer.optimize_graph(g, catch_errors) + onnx_graph = optimizer.optimize_graph(g, catch_errors, optimizers=optimizers) model_proto = onnx_graph.make_model("converted from {}".format(name), external_tensor_storage=external_tensor_storage) if output_path: