tensorflow-core-framework-graph_to_functiondef.h 2019-06-19 899 tensorflow-core-framework ```cpp #ifndef TENSORFLOW_CORE_FRAMEWORK_GRAPH_TO_FUNCTIONDEF_H_ #define TENSORFLOW_CORE_FRAMEWORK_GRAPH_TO_FUNCTIONDEF_H_ #include "absl/types/optional.h" #include "tensorflow/core/framework/function.h" #include "tensorflow/core/graph/graph.h" #include "tensorflow/core/lib/core/status.h" namespace tensorflow { // Converts 'graph' to a FunctionDef 'fdef', with name 'name': // // (1) 'node->IsArg()' nodes converted to function inputs. // (2) 'node->IsRetval()' nodes converted to function output. // (3) 'control_ret' returns an optional with a control output name, that will // be added to the function `control_ret` map (see FunctionDef) and // `control_output` in Op definition (see OpDef). Control output name must // be unique for all control output nodes. // // Closely modeled on the Python code in tensorflow/python/framework/function.py Status GraphToFunctionDef( const Graph& graph, const string& name, const std::function(const Node*)>& control_ret, FunctionDef* fdef); Status GraphToFunctionDef(const Graph& graph, const string& name, FunctionDef* fdef); } // namespace tensorflow #endif // TENSORFLOW_CORE_FRAMEWORK_GRAPH_TO_FUNCTIONDEF_H_ ``` 本文链接: http://www.codeeyes.net/archives/tensorflow-core-framework-graph_to_functiondef_h.html