How can I get the latency in the pi0 paper?

[lcf2610]

Hi everyone!
In pi0 paper, I found this latency table:

I want to know how can I get this by add code in the pi0 code. (because the "sample_actions" func is complied by XLA, the "print" and "time.time()" is useless)
Thanks!