Hi there,
I have gone in a bit of a loop lately trying to use the overlay as the example notebook display but using a tensorflow / keras model.
The notebook is great for tflite but this does not work if you provide a tensorboard profile input used from a tensorflow model serving capture.
I have tried a few things:

1. Try to parse the xplane.pb file but got some errors
2. Try to convert the tensorflow model to tensorflow light but then got some error due to flex support on the benchmark

Overall, I feel it would benefit to have some sort of tutorial on how to do it as this would be a great addition to the trace view in tensorboard profiler.

This kind of work requires deep understanding in the different tools so I think it might benefit from a colab.