From d50300a09fdb66ab882f4a44a239740cf533d321 Mon Sep 17 00:00:00 2001 From: gongdaguo Date: Fri, 13 Nov 2020 11:00:49 +0800 Subject: [PATCH] add demol --- mindspore/lite/src/ops/transpose.cc | 1 + mindspore/lite/src/runtime/kernel/arm/fp32/transpose_fp32.cc | 1 + mindspore/lite/test/models_only_for_process.cfg | 4 ++++ mindspore/lite/test/models_tflite.cfg | 2 ++ 4 files changed, 8 insertions(+) diff --git a/mindspore/lite/src/ops/transpose.cc b/mindspore/lite/src/ops/transpose.cc index 37bf1c4ab4..1c1dc46ec3 100644 --- a/mindspore/lite/src/ops/transpose.cc +++ b/mindspore/lite/src/ops/transpose.cc @@ -124,6 +124,7 @@ int Transpose::InferShape(std::vector inputs_, std::vector o if (!GetInferFlag()) { return RET_OK; } + MS_ASSERT(inputs_.size() == kSingleNum || inputs_.size() == kDoubleNum); MS_ASSERT(outputs_.size() == kSingleNum); int conjugate = GetConjugate(); diff --git a/mindspore/lite/src/runtime/kernel/arm/fp32/transpose_fp32.cc b/mindspore/lite/src/runtime/kernel/arm/fp32/transpose_fp32.cc index e9a736bede..08cf7df921 100644 --- a/mindspore/lite/src/runtime/kernel/arm/fp32/transpose_fp32.cc +++ b/mindspore/lite/src/runtime/kernel/arm/fp32/transpose_fp32.cc @@ -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(); diff --git a/mindspore/lite/test/models_only_for_process.cfg b/mindspore/lite/test/models_only_for_process.cfg index 3cd5271cdf..497dd57205 100644 --- a/mindspore/lite/test/models_only_for_process.cfg +++ b/mindspore/lite/test/models_only_for_process.cfg @@ -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 diff --git a/mindspore/lite/test/models_tflite.cfg b/mindspore/lite/test/models_tflite.cfg index 74aec76ed9..73ea6bcdb8 100644 --- a/mindspore/lite/test/models_tflite.cfg +++ b/mindspore/lite/test/models_tflite.cfg @@ -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