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() )。