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
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章