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