From 3541b22ce42ec457e64ccd4ece4204150a5b2b26 Mon Sep 17 00:00:00 2001
From: ysh329 <ysh329@users.noreply.github.com>
Date: Mon, 14 Sep 2020 11:32:50 +0800
Subject: [PATCH] fix opencl dropout. test=develop (#4253)

---
 lite/kernels/opencl/dropout_image_compute.cc | 1 +
 1 file changed, 1 insertion(+)

diff --git a/lite/kernels/opencl/dropout_image_compute.cc b/lite/kernels/opencl/dropout_image_compute.cc
index c3fdba3c13..c654129727 100644
--- a/lite/kernels/opencl/dropout_image_compute.cc
+++ b/lite/kernels/opencl/dropout_image_compute.cc
@@ -136,4 +136,5 @@ REGISTER_LITE_KERNEL(dropout,
                 {LiteType::GetTensorTy(TARGET(kOpenCL),
                                        PRECISION(kFP16),
                                        DATALAYOUT(kImageDefault))})
+    .BindOutput("Mask", {LiteType::GetTensorTy(TARGET(kARM))})
     .Finalize();
-- 
GitLab