Move ObjectIdentity to python/utils

PiperOrigin-RevId: 257637727

tensorflow/python/keras/engine/base_layer.py

@@ -67,11 +67,11 @@ from tensorflow.python.ops import variables as tf_variables
 from tensorflow.python.training.tracking import base as trackable
 from tensorflow.python.training.tracking import data_structures
 from tensorflow.python.training.tracking import layer_utils as trackable_layer_utils
-from tensorflow.python.training.tracking import object_identity
 from tensorflow.python.training.tracking import tracking
 from tensorflow.python.util import compat
 from tensorflow.python.util import deprecation
 from tensorflow.python.util import nest
+from tensorflow.python.util import object_identity
 from tensorflow.python.util import serialization
 from tensorflow.python.util import tf_decorator
 from tensorflow.python.util import tf_inspect

tensorflow/python/keras/layers/recurrent_test.py

@@ -46,9 +46,9 @@ from tensorflow.python.ops import special_math_ops
 from tensorflow.python.ops import state_ops
 from tensorflow.python.ops import variables as variables_lib
 from tensorflow.python.platform import test
-from tensorflow.python.training.tracking import object_identity
 from tensorflow.python.training.tracking import util as trackable_util
 from tensorflow.python.util import nest
+from tensorflow.python.util import object_identity
 
 # Used for nested input/output/state RNN test.
 NestedInput = collections.namedtuple('NestedInput', ['t1', 't2'])

tensorflow/python/keras/layers/wrappers_test.py

@@ -32,8 +32,8 @@ from tensorflow.python.keras.engine import base_layer_utils
 from tensorflow.python.keras.layers.rnn_cell_wrapper_v2 import ResidualWrapper
 from tensorflow.python.ops.array_ops import concat
 from tensorflow.python.platform import test
-from tensorflow.python.training.tracking import object_identity
 from tensorflow.python.training.tracking import util as trackable_util
+from tensorflow.python.util import object_identity
 
 
 class _RNNCellWithConstants(keras.layers.Layer):

tensorflow/python/saved_model/BUILD

@@ -311,7 +311,6 @@ py_library(
         "//tensorflow/python/training/tracking",
         "//tensorflow/python/training/tracking:base",
         "//tensorflow/python/training/tracking:graph_view",
-        "//tensorflow/python/training/tracking:object_identity",
         "//tensorflow/python/training/tracking:util",
     ],
 )

tensorflow/python/saved_model/save.py

@@ -51,10 +51,10 @@ from tensorflow.python.saved_model import utils_impl
 from tensorflow.python.training.saving import functional_saver
 from tensorflow.python.training.tracking import base
 from tensorflow.python.training.tracking import graph_view
-from tensorflow.python.training.tracking import object_identity
 from tensorflow.python.training.tracking import tracking
 from tensorflow.python.training.tracking import util
 from tensorflow.python.util import compat
+from tensorflow.python.util import object_identity
 from tensorflow.python.util.tf_export import tf_export
 
 _UNCOPIABLE_DTYPES = frozenset((dtypes.resource, dtypes.variant))

tensorflow/python/training/tracking/BUILD

@@ -103,19 +103,12 @@ tf_py_test(
     ],
 )
 
-py_library(
-    name = "object_identity",
-    srcs = ["object_identity.py"],
-    srcs_version = "PY2AND3",
-)
-
 py_library(
     name = "graph_view",
     srcs = ["graph_view.py"],
     srcs_version = "PY2AND3",
     deps = [
         ":base",
-        ":object_identity",
         ":tracking",
         "//tensorflow/core:protos_all_py",
         "//tensorflow/python:constant_op",
@@ -134,7 +127,6 @@ py_library(
         ":base",
         ":data_structures",
         ":graph_view",
-        ":object_identity",
         ":tracking",
         "//tensorflow/core:protos_all_py",
         "//tensorflow/python:array_ops",

tensorflow/python/training/tracking/graph_view.py

@@ -28,8 +28,8 @@ from tensorflow.python.training import optimizer as optimizer_v1
 from tensorflow.python.training.saving import saveable_object as saveable_object_lib
 from tensorflow.python.training.saving import saveable_object_util
 from tensorflow.python.training.tracking import base
-from tensorflow.python.training.tracking import object_identity
 from tensorflow.python.training.tracking import tracking
+from tensorflow.python.util import object_identity
 
 
 _ESCAPE_CHAR = "."  # For avoiding conflicts with user-specified names.

tensorflow/python/training/tracking/layer_utils.py

@@ -21,7 +21,7 @@ from __future__ import absolute_import
 from __future__ import division
 from __future__ import print_function
 
-from tensorflow.python.training.tracking import object_identity
+from tensorflow.python.util import object_identity
 
 
 def is_layer(obj):

tensorflow/python/training/tracking/util.py

@@ -49,11 +49,11 @@ from tensorflow.python.training.saving import saveable_object_util
 from tensorflow.python.training.tracking import base
 from tensorflow.python.training.tracking import data_structures
 from tensorflow.python.training.tracking import graph_view as graph_view_lib
-from tensorflow.python.training.tracking import object_identity
 from tensorflow.python.training.tracking import tracking
 from tensorflow.python.util import compat
 from tensorflow.python.util import deprecation
 from tensorflow.python.util import lazy_loader
+from tensorflow.python.util import object_identity
 from tensorflow.python.util import tf_contextlib
 from tensorflow.python.util.tf_export import tf_export