Browse Source

!14061 [lite]fix fill infer bug

From: @xu_anyue
Reviewed-by: @hangangqiang,@HilbertDavid
Signed-off-by: @hangangqiang
pull/14061/MERGE
mindspore-ci-bot Gitee 4 years ago
parent
commit
b141cb879c
3 changed files with 3 additions and 1 deletions
  1. +0
    -0
      mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/benchmark/load_input.c
  2. +0
    -0
      mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/benchmark/load_input.h
  3. +3
    -1
      mindspore/lite/nnacl/infer/fill_infer.c

+ 0
- 0
mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/benchmark/load_input.c View File


+ 0
- 0
mindspore/lite/micro/example/mnist_stm32f746/mnist_stm32f746/benchmark/load_input.h View File


+ 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_) {
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++) {


Loading…
Cancel
Save