Callback API to trace each operator?

TensorFlow
1

Hello, I’m new to TensorFlow.

I’m wondering if there’s any existing callback API for users to trace each operator in TensorFlow ? For example, detailing aspects such as the operator’s name, the number of inputs, and the input values, and the output.

3

Hi @Liang_Zhou, Could you please take a look at this Tensorflow profiler document which might be helpful for your use case. Thank You.