如何在TensorFlow中分析图形函数？

Question

在TensorFlow Eager中，我可以使用Python的profiler来分析纯粹在Eager模式下运行的代码。但是，如果我使用tf.function或tf.contrib.eager.defun将一个python函数“编译”为一个graph函数，则该函数对python来说是不透明的—探查器无法输入它。你知道吗

我已经知道如何在graph模式下分析TF图，但是我不知道如何在eager模式下使用graph函数。你知道吗

具体来说，如果我构造一个这样的函数

tf.enable_v2_behavior()

@tf.function
def myfunc(x):
  y = op1(x)
  z = op2(y, z)
  return z

当我执行myfunc时，如何计算在op1和op2中花费了多少时间？你知道吗