|
|
|
@@ -65,16 +65,6 @@ class GraphToFunctionDef { |
|
|
|
static void AddNodeAttr(const string &attr_name, |
|
|
|
const domi::tensorflow::AttrValue &value, |
|
|
|
domi::tensorflow::NodeDef *node_def); |
|
|
|
private: |
|
|
|
static domi::Status DataOpToFunctionDef(const ge::NodePtr &node, FunctionDef *fdef, |
|
|
|
std::map<string, string> &tensor_renaming, NameMapHelper &node_names); |
|
|
|
|
|
|
|
|
|
|
|
static domi::Status NetoutputOpToFunctionDef(const ge::NodePtr &node, FunctionDef *fdef, |
|
|
|
NameMapHelper &node_names); |
|
|
|
|
|
|
|
static domi::Status UpdateFunctionDef(const ge::NodePtr &node, NodeDef *node_def, |
|
|
|
NameMapHelper &node_names); |
|
|
|
}; |
|
|
|
|
|
|
|
class NameMapHelper { |
|
|
|
|