sbjkx
/
mindspore
forked from OSSInnovation/mindspore
1
0
Fork 0
Code Issues Pull Requests Releases Wiki Activity

!6418 Add available preset MindSpore model - shufflenetv2 

Merge pull request !6418 from XianglongZeng/myms
This commit is contained in:
mindspore-ci-bot 2020-09-18 16:19:42 +08:00 committed by Gitee
parent 7d2da469a9 3bd69c3961
commit faa0a6ad45
2 changed files with 5 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
mindspore/lite/src/ops/primitive_c.cc
View File

@ -446,6 +446,9 @@ std::shared_ptr<PrimitiveC> PrimitiveC::Create(const Primitive &prim, const std:
return NewPrimitiveC<BNGrad>(prim, inputs, quantType); return NewPrimitiveC<BNGrad>(prim, inputs, quantType);
} else if (op_type == "Tile") { } else if (op_type == "Tile") {
return NewPrimitiveC<Tile>(prim, inputs, quantType); return NewPrimitiveC<Tile>(prim, inputs, quantType);
#else
} else if (op_type == "Conv2DBackpropInput") {
return NewPrimitiveC<DeConv2D>(prim, inputs, quantType);
#endif #endif
} else { } else {
MS_LOG(ERROR) << "Unsupported primitive type in Create : " << op_type; MS_LOG(ERROR) << "Unsupported primitive type in Create : " << op_type;

1
mindspore/lite/test/models_mindspore.cfg
View File

@ -1,3 +1,4 @@
ssd.mindir ssd.mindir
mobilenetv2_438.mindir mobilenetv2_438.mindir
gate_u_net_small-1_110.mindir gate_u_net_small-1_110.mindir
shufflenetv2.mindir