diff --git a/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/benchmark/load_input.c b/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/benchmark/load_input.c old mode 100755 new mode 100644 diff --git a/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/benchmark/load_input.h b/mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/benchmark/load_input.h old mode 100755 new mode 100644 diff --git a/mindspore/lite/nnacl/infer/fill_infer.c b/mindspore/lite/nnacl/infer/fill_infer.c index 8359cdda46..0a357b34eb 100644 --- a/mindspore/lite/nnacl/infer/fill_infer.c +++ b/mindspore/lite/nnacl/infer/fill_infer.c @@ -38,7 +38,9 @@ int FillInferShape(const TensorC *const *inputs, size_t inputs_size, TensorC **o if (!parameter->infer_flag_) { return NNACL_INFER_INVALID; } - + if (num_dims != 0 && dst_shape == NULL) { + return NNACL_INFER_INVALID; + } int output_shape[MAX_SHAPE_SIZE]; size_t output_shape_size = 0; for (size_t i = 0; i < num_dims; i++) {