TVM python中relay.build流程

在用TVM優化過程中,常見demo如下圖: relay.build背後的調用機制梳理如下: python中的build函數實現如下,此方法實際調用了BuildModule類的build方法, BuildModule獲取的C++函數build方法,此處涉及混合編程: _build_module._BuildModule() 來自於這句註冊: tvm._ffi._init_api("relay.bui
相關文章
相關標籤/搜索