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);
GE_CHK_STATUS_RET(CheckOpType(node_def, op_type), "Failed to check op type");
} else {
op_type = ge::parser::FRAMEWORKOP;
domi::tensorflow::AttrValue attr_call_inference;
if ((node_name == node_op) &&
ge::TensorFlowUtil::FindAttrValue(node_def, "_disable_call_shape_inference", attr_call_inference)) {
op_type = node_op;
}
}



Loading…
Cancel
Save