Browse Source

support error_manager used in train

pull/249/head
wangxiaotian22 4 years ago
parent
commit
52354ca99f
2 changed files with 5 additions and 1 deletions
  1. +1
    -1
      metadef
  2. +4
    -0
      parser/tensorflow/tensorflow_parser.cc

+ 1
- 1
metadef

@@ -1 +1 @@
Subproject commit f982caa0981b1fdcc55a8ec27b4f4de9c58d33ba
Subproject commit 3df578f6d08e51e2c4ed8a023fb8482109941665

+ 4
- 0
parser/tensorflow/tensorflow_parser.cc View File

@@ -818,6 +818,8 @@ Status TensorFlowModelParser::CheckOpType(const domi::tensorflow::NodeDef *node_
* @return ge::DataType
*/
ge::DataType TensorFlowModelParser::ConvertToGeDataType(const uint32_t type) {
ErrorManager::GetInstance().GenWorkStreamIdDefault();

ge::DataType data_type = domi::TensorAssign::ConvertTensorflowDataType(type);
return data_type;
}
@@ -2139,6 +2141,7 @@ Status TensorFlowWeightsParser::ParseFromMemory(const char *data, uint32_t size,
Status TensorFlowWeightsParser::Parse(const char *file, ge::Graph &graph) { return SUCCESS; }

Status TensorFlowModelParser::ParseProto(const google::protobuf::Message *proto, ge::ComputeGraphPtr &graph) {
ErrorManager::GetInstance().GenWorkStreamIdDefault();
PARSER_TIMESTAMP_START(ParseProto);
GE_CHECK_NOTNULL(proto);
GE_CHECK_NOTNULL(graph);
@@ -2265,6 +2268,7 @@ Status TensorFlowModelParser::ParseProto(const google::protobuf::Message *proto,

Status TensorFlowModelParser::ParseProtoWithSubgraph(const google::protobuf::Message *root_proto,
domi::GetGraphCallback callback, ge::ComputeGraphPtr &root_graph) {
ErrorManager::GetInstance().GenWorkStreamIdDefault();
GE_CHECK_NOTNULL(root_proto);
GE_CHECK_NOTNULL(callback);
GE_CHECK_NOTNULL(root_graph);


Loading…
Cancel
Save