Write a script to benchmark inference latency of a pretrained model running on a TPU

Question

Can you write a script to benchmark inference latency of a pretrained model running on a TPU.