Simple example for the colored Compute time and Memory under Graphs tab?

Hi, I’m working on extending the Graphs plugin for PyTorch models and want to study how the color gradient feature for Compute time and Memory works. Can someone please share with me a super simple example of this feature in action? It’d be really nice if it’s code I can run in Google Colab, and the model in the example can be a Tensorflow model. I Googled but couldn’t find a working example, unfortunately.

Hi @Olivia_Liu, You can use the TensorFlow Profiler to profile the execution of your TensorFlow code. Please refer to this document for working code example. Thank You.

1 Like