提交 5a3a4b7d 编写于 作者: Z zhupengyang

Merge branch 'zhupy/fix-dropout' into 'incubate/lite'

fix arm kernel for dropout

See merge request inference/paddlelite!54
...@@ -43,4 +43,5 @@ REGISTER_LITE_KERNEL(dropout, kARM, kFloat, kNCHW, ...@@ -43,4 +43,5 @@ REGISTER_LITE_KERNEL(dropout, kARM, kFloat, kNCHW,
paddle::lite::kernels::arm::DropoutCompute, def) paddle::lite::kernels::arm::DropoutCompute, def)
.BindInput("X", {LiteType::GetTensorTy(TARGET(kARM))}) .BindInput("X", {LiteType::GetTensorTy(TARGET(kARM))})
.BindOutput("Out", {LiteType::GetTensorTy(TARGET(kARM))}) .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kARM))})
.BindOutput("Mask", {LiteType::GetTensorTy(TARGET(kARM))})
.Finalize(); .Finalize();
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册