fix arm kernel for dropout

test=develop

fix arm kernel for dropout
test=develop
a213f441 · zhupengyang · 1fbd77d4 · a213f441
隐藏空白更改
内联并排

Showing with 1 addition and 0 deletion

paddle/fluid/lite/kernels/arm/dropout_compute.cc paddle/fluid/lite/kernels/arm/dropout_compute.cc +1 -0

未找到文件。
--- a/paddle/fluid/lite/kernels/arm/dropout_compute.cc
+++ b/paddle/fluid/lite/kernels/arm/dropout_compute.cc
@@ -43,4 +43,5 @@ REGISTER_LITE_KERNEL(dropout, kARM, kFloat, kNCHW,
                     paddle::lite::kernels::arm::DropoutCompute, def)
    .BindInput("X", {LiteType::GetTensorTy(TARGET(kARM))})
    .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kARM))})
+    .BindOutput("Mask", {LiteType::GetTensorTy(TARGET(kARM))})
    .Finalize();