Commit b23b244c authored by Eric Liu's avatar Eric Liu Committed by TensorFlower Gardener
Browse files

[tpu:profiler] Support the Op Profile tool in TPU profiler. 

o Add an op_profile proto that defines a Profile class which assembles a
hierarchical performance profile based on HLOs in trace_events.
o Dump JSON-formatted op profile proto to the log directory.

PiperOrigin-RevId: 166318667
parent 410fe2c3
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment