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