From a213f441819a82b60ee2a375b891768e0c1607c1 Mon Sep 17 00:00:00 2001
From: zhupengyang <zhupengyang@baidu.com>
Date: Mon, 24 Jun 2019 05:32:01 +0000
Subject: [PATCH] fix arm kernel for dropout test=develop

---
 paddle/fluid/lite/kernels/arm/dropout_compute.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/paddle/fluid/lite/kernels/arm/dropout_compute.cc b/paddle/fluid/lite/kernels/arm/dropout_compute.cc
index 99b3e85974..49e998536a 100644
--- 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();
-- 
GitLab