Python tf.profiler.experimental.Trace.set_metadata用法及代碼示例

用法

set_metadata(
    **kwargs
)

在此跟蹤事件中設置元數據。

此方法允許在創建跟蹤事件後設置元數據。

示例用法：

def call(function):
    with tf.profiler.experimental.Trace("call",
         function_name=function.name) as tm:
      binary, in_cache = jit_compile(function)
      tm.set_metadata(in_cache=in_cache)
      execute(binary)

在這個例子中，我們想要跟蹤調用一個函數花費了多少時間，包括編譯和執行。編譯可以是獲取二進製文件的緩存副本，也可以實際生成二進製文件，這由 jit_compile() 返回的布爾值 "in_cache" 指示。我們需要使用set_metadata() 來傳遞in_cache，因為在創建跟蹤時我們不知道in_cache 的值(並且我們無法在jit_compile() 之後創建跟蹤，因為我們要測量整個持續時間call() )。