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);
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