Browse Source

!13878 [MS_LITE] fix issue to r1.2

From: @YeFeng_24
Reviewed-by: @hangangqiang,@zhanghaibo5
Signed-off-by: @hangangqiang
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
86d3a8eea0
1 changed files with 3 additions and 1 deletions
  1. +3
    -1
      mindspore/lite/nnacl/infer/fill_infer.c

+ 3
- 1
mindspore/lite/nnacl/infer/fill_infer.c View File

@@ -38,7 +38,9 @@ int FillInferShape(const TensorC *const *inputs, size_t inputs_size, TensorC **o
if (!parameter->infer_flag_) { if (!parameter->infer_flag_) {
return NNACL_INFER_INVALID; return NNACL_INFER_INVALID;
} }

if (num_dims != 0 && dst_shape == NULL) {
return NNACL_INFER_INVALID;
}
int output_shape[MAX_SHAPE_SIZE]; int output_shape[MAX_SHAPE_SIZE];
size_t output_shape_size = 0; size_t output_shape_size = 0;
for (size_t i = 0; i < num_dims; i++) { for (size_t i = 0; i < num_dims; i++) {


Loading…
Cancel
Save