diff --git a/tensorflow/python/grappler/cost_analyzer_tool.py b/tensorflow/python/grappler/cost_analyzer_tool.py
index 7dbaf449cad..92ff471de58 100644
--- a/tensorflow/python/grappler/cost_analyzer_tool.py
+++ b/tensorflow/python/grappler/cost_analyzer_tool.py
@@ -21,6 +21,8 @@ from __future__ import print_function
 import argparse
 import sys
 
+from absl import app
+
 from google.protobuf import message
 from google.protobuf import text_format
 from tensorflow.contrib.fused_conv.ops import gen_fused_conv2d_bias_activation_op  # pylint: disable=unused-import
@@ -32,7 +34,6 @@ from tensorflow.python.framework import importer
 from tensorflow.python.framework import ops
 from tensorflow.python.grappler import cost_analyzer
 from tensorflow.python.grappler import tf_optimizer
-from tensorflow.python.platform import app
 from tensorflow.python.platform import gfile
 from tensorflow.python.training import saver
 
diff --git a/tensorflow/python/grappler/graph_analyzer.py b/tensorflow/python/grappler/graph_analyzer.py
index c46a74ea64c..2e6dc640388 100644
--- a/tensorflow/python/grappler/graph_analyzer.py
+++ b/tensorflow/python/grappler/graph_analyzer.py
@@ -25,8 +25,9 @@ from __future__ import print_function
 import argparse
 import sys
 
+from absl import app
+
 from tensorflow.python import _pywrap_graph_analyzer as tf_wrap
-from tensorflow.python.platform import app
 
 
 def main(_):