srkreddy1238 commented on a change in pull request #5617:
URL: https://github.com/apache/incubator-tvm/pull/5617#discussion_r432813134
##########
File path: python/tvm/relay/frontend/tensorflow.py
##########
@@ -2773,6 +2774,12 @@ def from_tensorflow(self, graph, layout="NHWC",
shape=None, outputs=None):
if freezed_ops:
raise Exception("Graph is not frozen. Provide a frozen graph. "
"Found operators {}".format(freezed_ops))
+ stateful_ops = [op for op in missing_operators
+ if op in self._main_graph_proto._stateful_ops_list]
+ if stateful_ops:
+ raise Exception("Found stateful operators in this graph {}. " \
Review comment:
Again this will display exception with only stateful missing ops
(Doesn't show normal missing ops.).
I don't think we need separate list for stateful missing ops. Just add them
to missing_operators list and display as one.
----------------------------------------------------------------
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.
For queries about this service, please contact Infrastructure at:
[email protected]
