Browse Source

!386 回退代码

Merge pull request !386 from 赵鲁鹏/c79_open
r1.6.0
王涛 Gitee 4 years ago
parent
commit
87f695e2cf
1 changed files with 2 additions and 0 deletions
  1. +2
    -0
      parser/tensorflow/tensorflow_parser.cc

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

@@ -1049,9 +1049,11 @@ Status TensorFlowModelParser::AdaptOpType(const domi::tensorflow::NodeDef *node_
op_type = tensorflow_train_op_map.at(node_op); op_type = tensorflow_train_op_map.at(node_op);
GE_CHK_STATUS_RET(CheckOpType(node_def, op_type), "Failed to check op type"); GE_CHK_STATUS_RET(CheckOpType(node_def, op_type), "Failed to check op type");
} else { } else {
op_type = ge::parser::FRAMEWORKOP;
domi::tensorflow::AttrValue attr_call_inference; domi::tensorflow::AttrValue attr_call_inference;
if ((node_name == node_op) && if ((node_name == node_op) &&
ge::TensorFlowUtil::FindAttrValue(node_def, "_disable_call_shape_inference", attr_call_inference)) { ge::TensorFlowUtil::FindAttrValue(node_def, "_disable_call_shape_inference", attr_call_inference)) {
op_type = node_op;
} }
} }




Loading…
Cancel
Save