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