[BUG FIX] Fix write_to_array OP

2090d0f9 · huzhiqiang · GitHub · 7698abbd · 2090d0f9
隐藏空白更改
内联并排

Showing with 1 addition and 1 deletion

lite/kernels/arm/write_to_array_compute.cc lite/kernels/arm/write_to_array_compute.cc +1 -1

未找到文件。
--- a/lite/kernels/arm/write_to_array_compute.cc
+++ b/lite/kernels/arm/write_to_array_compute.cc
@@ -65,6 +65,6 @@ REGISTER_LITE_KERNEL(write_to_array,
                     paddle::lite::kernels::arm::WriteToArrayCompute,
                     def)
    .BindInput("X", {LiteType::GetTensorTy(TARGET(kARM), PRECISION(kAny))})
-    .BindInput("I", {LiteType::GetTensorTy(TARGET(kARM))})
+    .BindInput("I", {LiteType::GetTensorTy(TARGET(kARM), PRECISION(kAny))})
    .BindOutput("Out", {LiteType::GetTensorListTy(TARGET(kARM))})
    .Finalize();