python-3.x - TensorFlow tf.function 和 jit_compile

Question

我使用 Gauss-Kronrod 算法使用 Tensorflow 编写了数值积分函数。我注意到，当我用它来装饰它时，与仅@tf.function(jit_compile=True)相比它需要更长的时间来执行@tf.function。我脑海中浮现的唯一解释是它多次重新编译，但我不知道这是什么原因。该函数具有以下签名：

@tf.function(jit_compile=True)
def integrate(func, lower_interval, upper_interval):
    ...

它只被调用一次，我们想要集成的函数由 给出，func因为它也是编译的。另外，我尝试使用 TensorBoard Profiler，但它没有显示任何信息。它在性能摘要页面上显示所有指标的 0 毫秒。

Compilation Time     0ms
Output Time     0ms
etc.

我也尝试了该experimental_get_compiler_ir方法，但我不明白我应该看什么。任何可能导致此类问题或如何调试它的建议将不胜感激。

标签： python-3.xtensorflowtensorflow2.0