[GitHub] [tvm] leandron commented on a change in pull request #9562: Improve the keras frontend to support tflite 2.6

[GitBox <gi...@apache.org>]

leandron commented on a change in pull request #9562:
URL: https://github.com/apache/tvm/pull/9562#discussion_r755389593



##########
File path: python/tvm/relay/frontend/keras.py
##########
@@ -1266,11 +1266,15 @@ def _convert_layer(keras_layer, etab, scope=""):
             # - In Keras, model._network_nodes contains keys of all nodes relevant to the
             #   current model;
             # - In tf.Keras, this is already done as part of tensorflow.keras.network.get_config
-            if (
-                not is_tf_keras
-                and not model._node_key(keras_layer, node_idx) in model._network_nodes
-            ):
-                continue
+            if not is_tf_keras:
+                if (

Review comment:
       Can you put a comment on which branch refers to which version, e.g `# TFlite 2.6` ?




-- 
This is an automated message from the Apache Git Service.
To respond to the message, please log on to GitHub and use the
URL above to go to the specific comment.

To unsubscribe, e-mail: commits-unsubscribe@tvm.apache.org

For queries about this service, please contact Infrastructure at:
users@infra.apache.org