message: Cleanup the TODO item after the usage has been removed in OSS keras.
diff: @@ -27,9 +27,7 @@ from tensorflow.python.util.tf_export import tf_export
  # Whether the mixed precision graph rewrite has been enabled or not with
  # `enable_mixed_precision_graph_rewrite`. Used to turn on auto_mixed_precision
  # in ConfigProtos passed to Sessions.
  -# TODO(scottzhu): change to private attribute once the usage in OSS keras is
  -# removed.
  -mixed_precision_graph_rewrite_is_enabled = False
  +_mixed_precision_graph_rewrite_is_enabled = False
  # True if a Session has been created without the mixed precision graph rewrite
  @@ -40,19 +38,17 @@ _non_mixed_precision_session_created = False
  # Whether the global tf.keras.mixed_precision.Policy uses mixed precision. Used
  # to raise an error message if both a mixed Policy and the graph rewrite are
  # used at the same time.
  -# TODO(scottzhu): change to private attribute once the usage in OSS keras is
  -# removed.
  -using_mixed_precision_policy = False
  +_using_mixed_precision_policy = False
  @tf_export('__internal__.train.is_mixed_precision_graph_rewrite_enabled', v1=[])
  def is_mixed_precision_graph_rewrite_enabled():
  - return mixed_precision_graph_rewrite_is_enabled
  + return _mixed_precision_graph_rewrite_is_enabled
  def set_mixed_precision_graph_rewrite_enabled(enabled):
  - global mixed_precision_graph_rewrite_is_enabled
  - mixed_precision_graph_rewrite_is_enabled = enabled
  + global _mixed_precision_graph_rewrite_is_enabled
  + _mixed_precision_graph_rewrite_is_enabled = enabled
  def non_mixed_precision_session_created():
  @@ -65,10 +61,10 @@ def set_non_mixed_precision_session_created(created):
  def is_using_mixed_precision_policy():
  - return using_mixed_precision_policy
  + return _using_mixed_precision_policy
  @tf_export('__internal__.train.set_using_mixed_precision_policy', v1=[])
  def set_using_mixed_precision_policy(is_using):
  - global using_mixed_precision_policy
  - using_mixed_precision_policy = is_using
  + global _using_mixed_precision_policy
  + _using_mixed_precision_policy = is_using
  