How to read TPU consumption during inferencing on edge devices?

I am running some edge tpu model on edge tpu devices. Is there anyway I can read how much memory from TPU is consuming during the run?