fix onnx parser bug

pull/9165/head
yvette 4 years ago
parent e211e11b48
commit ff3d781916

@ -38,6 +38,7 @@ STATUS OnnxReduceParser::Parse(const onnx::GraphProto &onnx_graph, const onnx::N
return RET_NULL_PTR;
}
attr->keepDims = 1;
for (const auto &onnx_node_attr : onnx_node.attribute()) {
const auto &attribute_name = onnx_node_attr.name();
if (attribute_name == "axes") {
@ -58,6 +59,10 @@ STATUS OnnxReduceParser::Parse(const onnx::GraphProto &onnx_graph, const onnx::N
attr->mode = schema::ReduceMode_ReduceMin;
} else if (type == "ReduceSum") {
attr->mode = schema::ReduceMode_ReduceSum;
} else if (type == "ReduceProd") {
attr->mode = schema::ReduceMode_ReduceProd;
} else if (type == "ReduceSumSquare") {
attr->mode = schema::ReduceMode_ReduceSumSquare;
} else {
MS_LOG(ERROR) << "unsupported type";
return RET_ERROR;

@ -86,6 +86,7 @@ STATUS OnnxPReluParser::Parse(const onnx::GraphProto &onnx_graph, const onnx::No
}
}
if (!params.empty()) {
const onnx::TensorProto *slope = &params[0];
if (slope == nullptr) {
MS_LOG(ERROR) << "input error: params[0] is null";
@ -104,6 +105,9 @@ STATUS OnnxPReluParser::Parse(const onnx::GraphProto &onnx_graph, const onnx::No
return RET_ERROR;
}
}
} else {
MS_LOG(WARNING) << "The slope pf prelu is null, which may cause errors.";
}
op->primitive->value.type = schema::PrimitiveType_PReLU;
op->primitive->value.value = attr.release();

Loading…
Cancel
Save