使用Bert代码，我看到下面的导入行：

from tensorflow.compiler.xla.experimental.xla_sharding import xla_sharding

这导致了这个错误：

ModuleNotFoundError: No module named 'tensorflow.compiler.xla.experimental'

下面你可以看到我安装的版本和包：

$ pip list | grep tensorflow
tensorflow                        2.12.0
tensorflow-addons                 0.20.0
tensorflow-estimator              2.12.0
tensorflow-io-gcs-filesystem      0.32.0

$ find python3.8/site-packages/ -name compiler
python3.8/site-packages/google/protobuf/compiler
python3.8/site-packages/tensorflow/python/compiler
python3.8/site-packages/tensorflow/xla_aot_runtime_src/tensorflow/compiler
python3.8/site-packages/tensorflow/compiler
python3.8/site-packages/tensorflow/include/google/protobuf/compiler
python3.8/site-packages/tensorflow/include/tensorflow/compiler
python3.8/site-packages/jedi/third_party/typeshed/third_party/2and3/google/protobuf/compiler

$ python3 --version
Python 3.8.10

一切看起来都很好，但我想知道到底是什么问题。编译器似乎是在TF-2.4.0中引入的，所以2.12.0应该可以。
我试着一步一步来

>>> from tensorflow.compiler import xla
>>> from tensorflow.compiler.xla import experimental
Traceback (most recent call last):
  File "<stdin>", line 1, in <module>
ImportError: cannot import name 'experimental' from 'tensorflow.compiler.xla' (/home/mahmood/.local/lib/python3.8/site-packages/tensorflow/compiler/xla/__init__.py)

你知道吗？