From 7f09ecdf30523d8d15b6e6e8e3255d2b207f6f91 Mon Sep 17 00:00:00 2001 From: dongduo Date: Thu, 17 Dec 2020 19:20:17 +0800 Subject: [PATCH 1/2] Fix empty bug --- .../tensorflow_auto_mapping_parser_adapter.cc | 11 +++++++++++ 1 file changed, 11 insertions(+) diff --git a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc index 4d40e10..ebf8079 100644 --- a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc +++ b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc @@ -56,6 +56,17 @@ Status TensorFlowAutoMappingParserAdapter::ParseParams(const Message *op_src, ge } op.BreakConnect(); + if (op_dest->GetType() == EMPTY) { + domi::tensorflow::AttrValue attr; + if (TensorFlowUtil::FindAttrValue(node, kShapeAttrDtype, attr)) { + ge::DataType data_type = domi::TensorAssign::ConvertTensorflowDataType(static_cast(attr.type())); + AttrUtils::SetInt(op_dest, kShapeAttrDtype, data_type); + GETLOGD("Get dtype:%d success.", data_type); + } else { + GELOGW("Get dtype failed!"); + } + } + // add dynamic input/output if (op_dest->GetType() == IDENTITYN) { uint32_t dynamic_tensor_num = 0; From 39d8a838a8709537d0a540c3c09d9353db8ab108 Mon Sep 17 00:00:00 2001 From: dongduo Date: Thu, 17 Dec 2020 19:27:08 +0800 Subject: [PATCH 2/2] Fix empty bug --- parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc index ebf8079..2c4ad95 100644 --- a/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc +++ b/parser/tensorflow/tensorflow_auto_mapping_parser_adapter.cc @@ -61,7 +61,7 @@ Status TensorFlowAutoMappingParserAdapter::ParseParams(const Message *op_src, ge if (TensorFlowUtil::FindAttrValue(node, kShapeAttrDtype, attr)) { ge::DataType data_type = domi::TensorAssign::ConvertTensorflowDataType(static_cast(attr.type())); AttrUtils::SetInt(op_dest, kShapeAttrDtype, data_type); - GETLOGD("Get dtype:%d success.", data_type); + GELOGD("Get dtype:%d success.", data_type); } else { GELOGW("Get dtype failed!"); }