|
|
|
@@ -3331,7 +3331,7 @@ Status TensorFlowModelParser::OptimizeConstNodes4CustomOp(domi::tensorflow::Grap |
|
|
|
// solve the problem of protobuf index less current_size. |
|
|
|
GE_IF_BOOL_EXEC(current_node->input_size() == 0, GELOGI("Input size is 0, already optimized"); continue); |
|
|
|
|
|
|
|
if (it.moveType == domi::OMG_REMOVE_TYPE_WITH_COND) { |
|
|
|
if (it.moveType == domi::RemoveInputType::OMG_REMOVE_TYPE_WITH_COND) { |
|
|
|
domi::tensorflow::AttrValue attr_value; |
|
|
|
GE_IF_BOOL_EXEC(!(ge::TensorFlowUtil::FindAttrValue(current_node, it.attrName, attr_value)), |
|
|
|
REPORT_INNER_ERROR("E19999", "Op:%s register AttrName[%s] has no value, check invalid", |
|
|
|
@@ -3339,10 +3339,10 @@ Status TensorFlowModelParser::OptimizeConstNodes4CustomOp(domi::tensorflow::Grap |
|
|
|
GELOGE(INTERNAL_ERROR, "AttrName[%s] has no value!", it.attrName.c_str()); |
|
|
|
return PARAM_INVALID); |
|
|
|
GE_IF_BOOL_EXEC(attr_value.b() == it.attrValue, unused_inputs.insert(move_index)); |
|
|
|
} else if (it.moveType == domi::OMG_REMOVE_INPUT_WITH_ORIGINAL_TYPE && it.originalType == current_op_name) { |
|
|
|
} else if (it.moveType == domi::RemoveInputType::OMG_REMOVE_INPUT_WITH_ORIGINAL_TYPE && it.originalType == current_op_name) { |
|
|
|
GELOGD("Input %s:%d will be removed.", current_op_name.c_str(), move_index); |
|
|
|
unused_inputs.insert(move_index); |
|
|
|
} else if (it.moveType == domi::OMG_INPUT_REORDER) { |
|
|
|
} else if (it.moveType == domi::RemoveInputType::OMG_INPUT_REORDER) { |
|
|
|
auto inputs = current_node->input(); |
|
|
|
if (static_cast<size_t>(inputs.size()) != it.input_order.size()) { |
|
|
|
REPORT_INNER_ERROR("E19999", "Input size of node:%s(%s) is mismatched, new order size:%zu, input size:%d", |
|
|
|
|