Browse Source

add available preset MindSpore model - shufflenetv2

tags/v1.0.0
AGroupofProbiotocs 5 years ago
parent
commit
3bd69c3961
2 changed files with 5 additions and 1 deletions
  1. +3
    -0
      mindspore/lite/src/ops/primitive_c.cc
  2. +2
    -1
      mindspore/lite/test/models_mindspore.cfg

+ 3
- 0
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);
} 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;


+ 2
- 1
mindspore/lite/test/models_mindspore.cfg 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

Loading…
Cancel
Save