import_graph_def運行流水線

本來想看看import_graph_def能不能將某些variable改爲reuse的方式,一步一步跟蹤下去之後發現最終調用的是c_api裏的函數,貌似是不太好修改了,淺嘗輒止

檢查graph
檢查inputmap
檢查return elements
移除未知的默認屬性
修正input_map
獲取option
填充Options
導入graph
import_graph_def
get_registered_ops
_ProcessGraphDefParam
_ProcessInputMapParam
_ProcessReturnElementsParam
_RemoveDefaultAttrs
_ConvertInputMapValues
c_api_util.ScopedTFImportGraphDefOptions
_PopulateTFImportGraphDefOptions
c_api.TF_GraphImportGraphDefWithResults
c_api_util.ScopedTFImportGraphDefResults
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章