Explore Xprof Tools

This is a intial version of the exploration colab, more instructions to be added.

# Install the stable version of the profiler plugin
pip install -U tensorboard_plugin_profile
 
# git clone the xprof repo so we have access to the demo data there
git clone http://github.com/openxla/xprof
 
Cloning into 'xprof'...
warning: redirecting to https://github.com/openxla/xprof/
remote: Enumerating objects: 11215, done.
remote: Counting objects: 100% (1518/1518), done.
remote: Compressing objects: 100% (627/627), done.
remote: Total 11215 (delta 1123), reused 894 (delta 891), pack-reused 9697 (from 3)
Receiving objects: 100% (11215/11215), 51.85 MiB | 35.49 MiB/s, done.
Resolving deltas: 100% (8360/8360), done.

# Load the TensorBoard notebook extension.
%load_ext tensorboard

# Launch TensorBoard and navigate to the Profile tab to view performance profile
%tensorboard --logdir=xprof/demo

Once tensorboard loads the profile plugin, Use the tools drop down to explore the tools.