From 6e44a001fe63bd28e39a5847b958b834859851eb Mon Sep 17 00:00:00 2001 From: zhangwen31 Date: Thu, 24 Sep 2020 11:53:24 +0000 Subject: [PATCH] revert: add i64 elementwie_mul back --- lite/kernels/arm/elementwise_compute.cc | 10 ++++++++++ 1 file changed, 10 insertions(+) diff --git a/lite/kernels/arm/elementwise_compute.cc b/lite/kernels/arm/elementwise_compute.cc index f207c8cfe1..17a6cf1aac 100644 --- a/lite/kernels/arm/elementwise_compute.cc +++ b/lite/kernels/arm/elementwise_compute.cc @@ -441,6 +441,16 @@ REGISTER_LITE_KERNEL( .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kARM))}) .Finalize(); +using elementwise_mul_int64_t = + paddle::lite::kernels::arm::ElementwiseMulCompute; +REGISTER_LITE_KERNEL( + elementwise_mul, kARM, kInt64, kNCHW, elementwise_mul_int64_t, def) + .BindInput("X", {LiteType::GetTensorTy(TARGET(kARM), PRECISION(kInt64))}) + .BindInput("Y", {LiteType::GetTensorTy(TARGET(kARM), PRECISION(kInt64))}) + .BindOutput("Out", {LiteType::GetTensorTy(TARGET(kARM), PRECISION(kInt64))}) + .Finalize(); + using elementwise_mul_float_t = paddle::lite::kernels::arm::ElementwiseMulCompute; REGISTER_LITE_KERNEL( -- GitLab