[Serving] Fix syntax error in generate timeline tool.

Signed-off-by: candy.dc <candy.dc@alibaba-inc.com>
DeepRec-AI · Feb 28, 2024 · ec3c9d2 · ec3c9d2
1 parent 8b58f9b
commit ec3c9d2
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/serving/tools/timeline/gen_timeline.py b/serving/tools/timeline/gen_timeline.py
@@ -1,6 +1,6 @@
 import sys
-import config_pb2
-import timeline
+from tensorflow.core.protobuf import config_pb2
+from tensorflow.python.client import timeline
 
 def gen_timeline(src_name, dest_name):
   run_metadata = config_pb2.RunMetadata()