mindspore-Ecosystem
/
mindspore
21
0
Fork 27
Code Issues Pull Requests 6 Projects Releases Wiki Activity

!41112 Codex fix 

Merge pull request !41112 from 陈宇/dev_r1.8_codex_fix
This commit is contained in:
i-robot 2022-08-31 02:11:33 +00:00 committed by Gitee
parent 2f251a3b64 af9909534a
commit 531257d16a
No known key found for this signature in database
GPG Key ID: 173E9B9CA92EEF8F
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
mindspore/lite/tools/converter/parser/onnx/onnx_nonzero_parser.cc
View File

@ -24,9 +24,9 @@ namespace mindspore {
namespace lite {
PrimitiveCPtr OnnxNonZeroParser::Parse(const onnx::GraphProto &onnx_graph, const onnx::NodeProto &onnx_node) {
auto prim = std::make_unique<ops::Where>();
MS_CHECK_TRUE_RET(prim != nullptr, nullptr);
auto prim_c = prim->GetPrim();
prim_c->AddAttr("is_nonzero", MakeValue(true));
MS_CHECK_TRUE_RET(prim != nullptr, nullptr);
return prim->GetPrim();
}