Browse Source

modify tf_data_parser check

pull/214/head
gengchao4@huawei.com 5 years ago
parent
commit
a87922d654
1 changed files with 3 additions and 2 deletions
  1. +3
    -2
      parser/tensorflow/tensorflow_data_parser.cc

+ 3
- 2
parser/tensorflow/tensorflow_data_parser.cc View File

@@ -81,8 +81,9 @@ Status TensorFlowDataParser::ParseInputFromModel(const Message *op_src, ge::OpDe
}

if (!TensorFlowUtil::FindAttrValue(node, TENSORFLOW_ATTR_SHAPE, attr_value)) {
GELOGE(domi::PARAM_INVALID, "input data node %s do not find shape.", node->name().c_str());
return domi::PARAM_INVALID;
// in some case, data could be without shape and is updated by `input_shape` option in following process
GELOGW("input data node %s do not find shape.", node->name().c_str());
return SUCCESS;
}

// Check shape attribute must be shape


Loading…
Cancel
Save