diff --git a/lite/kernels/opencl/dropout_image_compute.cc b/lite/kernels/opencl/dropout_image_compute.cc index c3fdba3c1363141b5dec4a73fa86985120a1e48a..c654129727494d4c0308e21546663b5ab4ac6779 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();