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

Question

在 TensorFlow Eager 中，我可以使用 Python 的分析器来分析纯粹以 Eager 模式运行的代码。tf.function但是，如果我使用or将 python 函数“编译”为图形函数tf.contrib.eager.defun，则该函数对 python 变得不透明 - 探查器无法输入它。

我已经找到了如何在图形模式下分析 TF 图，但我不知道如何在渴望模式下使用图形函数来做到这一点。

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

tf.enable_v2_behavior()

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

如何确定执行 myfunc 的时间op1和时间op2？

标签： pythontensorflowprofilereager-execution