Browse Source

!347 回退 'Pull Request !345 : size of parser'

Merge pull request !347 from 王涛/revert-merge-345-master
r1.5.1
i-robot Gitee 3 years ago
parent
commit
09ca43bcd6
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc

+ 1
- 1
parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc View File

@@ -87,7 +87,7 @@ Status TensorFlowAutoMappingParserAdapter::ParseParams(const Message *op_src, ge
if (op_dest->GetType() == SIZE) { if (op_dest->GetType() == SIZE) {
ge::DataType out_type = DT_INT32; ge::DataType out_type = DT_INT32;
if (AttrUtils::GetDataType(op_dest, kShapeAttrOutType, out_type)) { if (AttrUtils::GetDataType(op_dest, kShapeAttrOutType, out_type)) {
if (!AttrUtils::SetDataType(op_dest, kShapeAttrDtype, out_type)) {
if (!AttrUtils::SetInt(op_dest, kShapeAttrDtype, static_cast<int64_t>(out_type))) {
REPORT_CALL_ERROR("E19999", "Set Attr:%s to op:%s(%s) failed", kShapeAttrDtype, REPORT_CALL_ERROR("E19999", "Set Attr:%s to op:%s(%s) failed", kShapeAttrDtype,
op_dest->GetName().c_str(), op_dest->GetType().c_str()); op_dest->GetName().c_str(), op_dest->GetType().c_str());
GELOGE(FAILED, "Set attr dtype for op:%s failed.", op_dest->GetName().c_str()); GELOGE(FAILED, "Set attr dtype for op:%s failed.", op_dest->GetName().c_str());


Loading…
Cancel
Save