Programmatically get operator information from profiler

I’d like to extract all the operator tracing info from a training job, such as operator name, shape, dtype, etc. Profiler can generate files like ip-xxx-xx-xx-xx.xplane.pb, and can be consumed by tensorboard to analyze through GUI. Is there an easy way to programmatically extract the info from these files?

Hi @KevinW ,

To extract operator tracing info from .xplane.pb files:

  • Load the trace using tf.profiler.experimental.load_profile.
  • Get a list of operators using trace.execution_stats.timeline.
  • Iterate over operators and extract information like name, shape, and dtype.

You can refer to the documentation for the same for better understanding and fuctionality .

Thank You