From 5559536470f81213c8f6460761654383d2cebf7b Mon Sep 17 00:00:00 2001 From: zhangyue <138768300+zhangyue207@users.noreply.github.com> Date: Sun, 28 Apr 2024 11:28:28 +0800 Subject: [PATCH] add kunlun squeeze kernel (#229) Co-authored-by: Haojie Wang --- src/kernels/kunlun/unary.cc | 1 + 1 file changed, 1 insertion(+) diff --git a/src/kernels/kunlun/unary.cc b/src/kernels/kunlun/unary.cc index 5fa2e99b..2df28f8c 100644 --- a/src/kernels/kunlun/unary.cc +++ b/src/kernels/kunlun/unary.cc @@ -570,6 +570,7 @@ REGISTER_KERNEL(Device::KUNLUN, OpType::Reciprocal, ReciprocalXdnn, REGISTER_KERNEL(Device::KUNLUN, OpType::Reshape, CopyXdnn, "Reshape_xdnn"); REGISTER_KERNEL(Device::KUNLUN, OpType::Flatten, CopyXdnn, "Flatten_xdnn"); REGISTER_KERNEL(Device::KUNLUN, OpType::Identity, CopyXdnn, "Identity_xdnn"); +REGISTER_KERNEL(Device::KUNLUN, OpType::Squeeze, CopyXdnn, "Squeeze_xdnn"); REGISTER_KERNEL(Device::KUNLUN, OpType::Abs, AbsXdnn, "Abs_xdnn"); REGISTER_KERNEL(Device::KUNLUN, OpType::Atan, ATanXdnn, "Atan_xdnn"); REGISTER_KERNEL(Device::KUNLUN, OpType::Log, LogXdnn, "Log_xdnn");