a fixing for gpt-2 subclassing model. (#368)

* remove tf2onnx api * migrate to tf2onnx * remove tf2onnx test cases and package. * minor corrections * update the version and docs * doc fixing * a fixing for gpt-2 subclassing model.
onnx · Jan 30, 2020 · 5df283e · 5df283e
1 parent 7ac8bbf
commit 5df283e
Showing 1 changed file with 1 addition and 0 deletions.
diff --git a/keras2onnx/_parse_tf.py b/keras2onnx/_parse_tf.py
@@ -223,6 +223,7 @@ def extract_outputs_from_inbound_nodes(model):
 
 def build_layer_output_from_model(model, output_dict, output_names):
     if is_subclassing(model):
+        tf.compat.v1.enable_tensor_equality()  # re-enable tensor tensor equality for subclassing model.
         return extract_outputs_from_subclassing_model(model, output_dict, output_names)
     else:
         graph = model.outputs[0].graph