Browse Source

backspace

pull/386/head
zhao-lupeng 3 years ago
parent
commit
7c00192dcb
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