torch.compiler API 參考¶
有關 torch.compiler 的快速概述,請參閱 torch.compiler。
有關此函式的引數詳情,請參閱 |
|
此函式清除所有編譯快取並將系統恢復到初始狀態。 |
|
告知編譯器前端 (Dynamo) 跳過函式的符號內省,而是在遇到函式時直接將其寫入圖中。 |
|
註冊一個函式的 polyfill 處理器,通常是來自 C 擴充套件的 C 函式,用於在圖中內聯原始函式時替代原始函式。 |
|
此函式用於將函式 fn 標記為具有常量結果。 |
|
返回可傳遞給 torch.compile(..., backend="name") 的有效字串。 |
|
此函式提供一個裝飾器來停用函式的編譯。 |
|
設定編譯器的當前立場。 |
|
指示推理或訓練的新迭代即將開始。 |
|
指示圖是否作為 |
|
指示圖是否透過 TorchDynamo 進行跟蹤。 |
|
指示是否正在匯出。 |