Browse Source

!8555 [MS][LITE]Add model

From: @gongdaguo
Reviewed-by: @zhanghaibo5,@hangangqiang
Signed-off-by: @zhanghaibo5,@hangangqiang
tags/v1.1.0
mindspore-ci-bot Gitee 5 years ago
parent
commit
8c6537e553
4 changed files with 8 additions and 0 deletions
  1. +1
    -0
      mindspore/lite/src/ops/transpose.cc
  2. +1
    -0
      mindspore/lite/src/runtime/kernel/arm/fp32/transpose_fp32.cc
  3. +4
    -0
      mindspore/lite/test/models_only_for_process.cfg
  4. +2
    -0
      mindspore/lite/test/models_tflite.cfg

+ 1
- 0
mindspore/lite/src/ops/transpose.cc View File

@@ -124,6 +124,7 @@ int Transpose::InferShape(std::vector<Tensor *> inputs_, std::vector<Tensor *> o
if (!GetInferFlag()) {
return RET_OK;
}
MS_ASSERT(inputs_.size() == kSingleNum || inputs_.size() == kDoubleNum);
MS_ASSERT(outputs_.size() == kSingleNum);

int conjugate = GetConjugate();


+ 1
- 0
mindspore/lite/src/runtime/kernel/arm/fp32/transpose_fp32.cc View File

@@ -116,6 +116,7 @@ int TransposeFp32Run(void *cdata, int task_id) {
}

int TransposeCPUKernel::Run() {
MS_ASSERT(in_tensors_.size() == 1 || in_tensors_.size() == 2);
MS_ASSERT(out_tensors_.size() == 1);
auto &in_tensor = in_tensors_.front();
auto &out_tensor = out_tensors_.front();


+ 4
- 0
mindspore/lite/test/models_only_for_process.cfg View File

@@ -2,6 +2,10 @@ lite-model_arbitrary-image-stylization-inceptionv3_dr_transfer_1.tflite
lite-model_arbitrary-image-stylization-inceptionv3_int8_transfer_1.tflite
lite-model_arbitrary-image-stylization-inceptionv3-dynamic-shapes_dr_transfer_1.tflite
lite-model_cartoongan_dr_1.tflite
mindspore_efficientnet_b0.mindir
mindspore_efficientnet_b4minus.mindir
mindspore_tinynet-a.mindir
mindspore_tinynet-e.mindir
lite-model_deeplabv3-mobilenetv2_1_default_1.tflite
lite-model_deeplabv3-mobilenetv2_dm05_1_default_1.tflite
lite-model_deeplabv3-mobilenetv2-int8_1_default_1.tflite


+ 2
- 0
mindspore/lite/test/models_tflite.cfg View File

@@ -108,6 +108,8 @@ mnasnet_1.0_160_1_metadata_1.tflite
mnasnet_1.0_192_1_metadata_1.tflite
mnasnet_1.0_224_1_metadata_1.tflite
mnasnet_1.0_96_1_metadata_1.tflite
ml_vision_guide_detection1.tflite
ml_vision_guide_detection3.tflite
lite-model_on_device_vision_classifier_popular_us_products_V1_1.tflite
lite-model_on_device_vision_classifier_popular_wine_V1_1.tflite
posenet_mobilenet_float_075_1_default_1.tflite


Loading…
Cancel
Save