fix opencl dropout. test=develop (#4253)

3541b22c · ysh329 · GitHub · d7c4af74 · 3541b22c
显示空白变更内容
内联并排

Showing with 1 addition and 0 deletion

lite/kernels/opencl/dropout_image_compute.cc lite/kernels/opencl/dropout_image_compute.cc +1 -0

未找到文件。
--- 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();