Remove legacy references from ops.py.

This is done to eventually remove the lazy loads in `indexed_slices.py`.

PiperOrigin-RevId: 504679098

Author: jam14j

mesh_tensorflow/tpu_variables.py

@@ -24,6 +24,7 @@
 
 # pylint: disable=g-direct-tensorflow-import
 from tensorflow.python.framework import ops
+from tensorflow.python.framework import tensor_conversion_registry
 from tensorflow.python.ops import control_flow_ops
 from tensorflow.python.ops import gen_resource_variable_ops
 
@@ -223,7 +224,8 @@ def _tensor_conversion(var, dtype=None, name=None, as_ref=False):
   return var._dense_var_to_tensor(dtype=dtype, name=name, as_ref=as_ref)  # pylint: disable=protected-access
 
 
-ops.register_tensor_conversion_function(ReplicatedVariable, _tensor_conversion)
+tensor_conversion_registry.register_tensor_conversion_function(
+    ReplicatedVariable, _tensor_conversion)
 
 if not TF_23:
   ops.register_dense_tensor_like_type(ReplicatedVariable)