Unit test failures on ARM_CI
Fixed by merge of #60347  Are you satisfied with the resolution of your issue?
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=Yes&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/60328">Yes</a>
<a href="https://docs.google.com/forms/d/e/1FAIpQLSfaP12TRhd9xSxjXZjcZFNXPGk4kc1-qMdv3gc6bEP90vY1ew/viewform?entry.85265664=No&entry.2137816233=https://github.com/tensorflow/tensorflow/issues/60328">No</a>

<details><summary>Click to expand!</summary> 
 
 ### Issue Type

Bug

### Have you reproduced the bug with TF nightly?

Yes

### Source

source

### Tensorflow Version

git HEAD

### Custom Code

No

### OS Platform and Distribution

Ubuntu 20.04

### Mobile device

n/a

### Python version

3.9.16

### Bazel version

5.3.0

### GCC/Compiler version

10.2.1

### CUDA/cuDNN version

n/a

### GPU model and memory

n/a

### Current Behaviour?

Since the commit https://github.com/tensorflow/tensorflow/commit/11c29fb2c3e508c3459b55d6357d8ff1d18412ea ARM_CI has been showing unit test failures on
//bazel_pip/tensorflow/python/compiler/xla:xla_test_gpu
//bazel_pip/tensorflow/python/data/experimental/kernel_tests:checkpoint_input_pipeline_hook_test
//bazel_pip/tensorflow/python/distribute:parameter_server_strategy_test_cpu
//bazel_pip/tensorflow/python/compiler/xla:xla_test_cpu
//bazel_pip/tensorflow/python/distribute:parameter_server_strategy_test_gpu
//bazel_pip/tensorflow/core/platform:ram_file_system_test

### Standalone code to reproduce the issue

```shell
bazel --bazelrc /usertools/aarch64.bazelrc test --config=mkl_aarch64_threadpool --copt=-flax-vector-conversions --test_env=TF_ENABLE_ONEDNN_OPTS=1 --test_env=TF2_BEHAVIOR=1 --define=tf_api_version=2 --flaky_test_attempts=3 --test_output=errors --verbose_failures=true --test_keep_going --jobs=75 --notest_verbose_timeout_warnings --build_tests_only -- //tensorflow/core/platform:ram_file_system_test //tensorflow/python/distribute:parameter_server_strategy_test_gpu //tensorflow/python/compiler/xla:xla_test_cpu //tensorflow/python/distribute:parameter_server_strategy_test_cpu //tensorflow/python/data/experimental/kernel_tests:checkpoint_input_pipeline_hook_test //tensorflow/python/compiler/xla:xla_test_gpu
```


### Relevant log output

```shell
All tests fail with similar backtrace

INFO: From Testing //bazel_pip/tensorflow/core/platform:ram_file_system_test:
  File "/tmpfs/bazel_output/_bazel_ubuntu/eab0d61a99b6696edb3d2aff87b585e8/execroot/org_tensorflow/bazel-out/aarch64-opt/bin/bazel_pip/tensorflow/core/platform/ram_file_system_test.runfiles/org_tensorflow/bazel_pip/tensorflow/core/platform/ram_file_system_test.py", line 21, in <module>
    from tensorflow.python.estimator.estimator import Estimator
  File "/tmpfs/bazel_output/_bazel_ubuntu/eab0d61a99b6696edb3d2aff87b585e8/execroot/org_tensorflow/bazel-out/aarch64-opt/bin/bazel_pip/tensorflow/core/platform/ram_file_system_test.runfiles/org_tensorflow/tensorflow/python/estimator/estimator.py", line 22, in <module>
    from tensorflow_estimator.python.estimator import estimator
  File "/workspace/pip_test/venv_clean/lib/python3.10/site-packages/tensorflow_estimator/__init__.py", line 8, in <module>
    from tensorflow_estimator._api.v1 import estimator
  File "/workspace/pip_test/venv_clean/lib/python3.10/site-packages/tensorflow_estimator/_api/v1/estimator/__init__.py", line 11, in <module>
    from tensorflow_estimator._api.v1.estimator import tpu
  File "/workspace/pip_test/venv_clean/lib/python3.10/site-packages/tensorflow_estimator/_api/v1/estimator/tpu/__init__.py", line 12, in <module>
    from tensorflow_estimator.python.estimator.tpu.tpu_estimator import TPUEstimator
  File "/workspace/pip_test/venv_clean/lib/python3.10/site-packages/tensorflow_estimator/python/estimator/tpu/tpu_estimator.py", line 118, in <module>
    to_proto=resource_variable_ops._to_proto_fn,  # pylint: disable=protected-access
AttributeError: module 'tensorflow.python.ops.resource_variable_ops' has no attribute '_to_proto_fn'
```
</details>