@@ -104,6 +104,8 @@ using concati64 =
104
104
paddle::lite::kernels::xpu::ConcatCompute<int64_t , PRECISION(kFloat )>;
105
105
using concati8 =
106
106
paddle::lite::kernels::xpu::ConcatCompute<int8_t , PRECISION(kInt8 )>;
107
+ using concatbool =
108
+ paddle::lite::kernels::xpu::ConcatCompute<bool , PRECISION(kFloat )>;
107
109
108
110
REGISTER_LITE_KERNEL (concat, kXPU , kFloat , kNCHW , concatfp32, def)
109
111
.BindInput(" X" , {LiteType::GetTensorTy (TARGET (kXPU ), PRECISION (kFloat ))})
@@ -147,3 +149,9 @@ REGISTER_LITE_KERNEL(concat, kXPU, kInt8, kNCHW, concati8, concat_INT8)
147
149
{LiteType::GetTensorTy (TARGET (kXPU ), PRECISION (kInt32 ))})
148
150
.BindOutput(" Out" , {LiteType::GetTensorTy (TARGET (kXPU ), PRECISION (kInt8 ))})
149
151
.Finalize();
152
+ REGISTER_LITE_KERNEL (concat, kXPU , kFloat , kNCHW , concatbool, concat_BOOL)
153
+ .BindInput(" X" , {LiteType::GetTensorTy (TARGET (kXPU ), PRECISION (kBool ))})
154
+ .BindInput(" AxisTensor" ,
155
+ {LiteType::GetTensorTy (TARGET (kXPU ), PRECISION (kInt32 ))})
156
+ .BindOutput(" Out" , {LiteType::GetTensorTy (TARGET (kXPU ), PRECISION (kBool ))})
157
+ .Finalize();
0 commit comments