I tried the tensorboard profiler in a Keras example. It doesn't work either. https://github.com/tensorflow/tensorboard/blob/master/docs/tensorboard_profiling_keras.ipynb

https://github.com/tensorflow/text/issues/538 https://github.com/tensorflow/text/issues/654

Understood. Any ETA on tensorflow_text?

ModuleNotFoundError: No module named 'tensorflow_text'

Hopefully, they'll use this for testing: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/python/profiler/profiler_v2_test.py

Python 3.8.5