Browse Source

!13731 [lite]fix fp16 gather bug

From: @xu_anyue
Reviewed-by: @hangangqiang,@zhanghaibo5
Signed-off-by: @hangangqiang
tags/v1.2.0-rc1
mindspore-ci-bot Gitee 5 years ago
parent
commit
28f953cfbd
1 changed files with 1 additions and 1 deletions
  1. +1
    -1
      mindspore/lite/src/runtime/kernel/arm/fp16/gather_fp16.cc

+ 1
- 1
mindspore/lite/src/runtime/kernel/arm/fp16/gather_fp16.cc View File

@@ -45,7 +45,7 @@ int GatherFp16CPUKernel::Init() {
reinterpret_cast<float16_t *>(context_->allocator->Malloc(input_tensor->ElementsNum() * sizeof(float16_t)));
Float32ToFloat16(reinterpret_cast<float *>(input_tensor->data_c()), input_data_, input_tensor->ElementsNum());
}
(reinterpret_cast<GatherParameter *>(op_parameter_))->axis_ = *static_cast<int *>(in_tensors_.at(2)->data_c());
if (!InferShapeDone()) {
return RET_OK;
}


Loading…
Cancel
Save