srkreddy1238 commented on a change in pull request #7498:
URL: https://github.com/apache/tvm/pull/7498#discussion_r584029723
##########
File path: python/tvm/relay/frontend/tensorflow.py
##########
@@ -3682,6 +3729,30 @@ def from_tensorflow(graph, layout="NHWC", shape=None,
outputs=None):
params : dict of str to tvm.nd.NDArray
Dict of converted parameters stored in tvm.nd.NDArray format
"""
+ from tensorflow.python.eager.function import ConcreteFunction
+
+ if isinstance(graph, ConcreteFunction):
+ try:
+ from tensorflow.python.framework import convert_to_constants
+ from tensorflow.core.protobuf import config_pb2
+ except ImportError as e:
+ raise ImportError("Unable to import tensorflow which is required
{}".format(e))
+ concrete_func = graph
+ graph =
convert_to_constants.convert_variables_to_constants_v2(concrete_func).graph
+ signature = _build_signature_def(graph, concrete_func.inputs,
concrete_func.outputs)
+ graph_def = graph.as_graph_def()
+
+ # Some optimization
+ config = config_pb2.ConfigProto()
+ rewriter_config = config.graph_options.rewrite_options
+ rewriter_config.optimizers[:] = [
+ "debug_stripper",
+ "arithmetic",
+ "dependency",
+ "arithmetic",
Review comment:
"arithmetic" is not a duplicate, this transform happen again.
The transforms order was with a ref. from TensorflowJS project. This can be
revised later as needed.
----------------------------------------------------------------
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]
