jax.profiler.start_server#

jax.profiler.start_server(port)[原始碼]#

port 埠口啟動 profiler 伺服器。

使用 TensorBoard 2.2 或更新版本中的「TensorFlow profiler」功能,您可以連線到 profiler 伺服器並取樣執行追蹤,以顯示 CPU、GPU 和/或 TPU 裝置活動。

參數:

port (int)

返回類型:

xla_client.profiler.ProfilerServer