add available preset MindSpore model - shufflenetv2

This commit is contained in:
AGroupofProbiotocs 2020-09-18 15:24:47 +08:00
parent 076d8ae530
commit 3bd69c3961
2 changed files with 5 additions and 1 deletions

View File

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

View File

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