I’m afraid you can’t. That would depend on the operation you’re performing. For example, for forward pass of convolution, you can find the list here 781.
You would have to add this print directly in the cpp code linked above.