Browse Source

clean code

pull/640/head
13291271729 3 years ago
parent
commit
616e871b36
2 changed files with 1 additions and 11 deletions
  1. +1
    -1
      parser/common/convert/message2operator.h
  2. +0
    -10
      parser/tensorflow/graph_functiondef.h

+ 1
- 1
parser/common/convert/message2operator.h View File

@@ -41,7 +41,7 @@ class Message2Operator {
static void ParseRepeatedBaseTypeField(const google::protobuf::Reflection *reflection,
const google::protobuf::Message *message,
const google::protobuf::FieldDescriptor *field,
ge::Operator &ops);
ge::Operator &ops, const int field_size);
};
} // namespace ge
#endif // PARSER_MESSAGE2OPERATOR_H

+ 0
- 10
parser/tensorflow/graph_functiondef.h View File

@@ -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 {


Loading…
Cancel
Save