[Serving] Fix syntax error in generate timeline tool. (#977)

Signed-off-by: candy.dc <[email protected]>
DeepRec-AI · Feb 29, 2024 · 186afd0 · 186afd0
1 parent 8b58f9b
commit 186afd0
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()