From e0194cf05644f2622cce6070a62ebd82f344b981 Mon Sep 17 00:00:00 2001 From: yefeng Date: Tue, 23 Mar 2021 20:14:19 +0800 Subject: [PATCH] 067-fix-issue_r1.2 --- mindspore/lite/nnacl/infer/fill_infer.c | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) 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++) {