forked from mindspore-Ecosystem/mindspore
add available preset MindSpore model - shufflenetv2
This commit is contained in:
parent
076d8ae530
commit
3bd69c3961
|
@ -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;
|
||||
|
|
|
@ -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
|
||||
|
|
Loading…
Reference in New Issue