InfiniTensor/test/kernels
Bolun Zhang 23b825efc4
Xpu task4 support: add softmax (#172)
* add softmax on kunlun

* format

---------

Co-authored-by: Bolun <bolunz@u.nus.edu>
Co-authored-by: Haojie Wang <haojie0429@gmail.com>
2023-10-30 16:01:05 +08:00
..
bang Pooling ceil mode (#155) 2023-10-09 20:51:39 +08:00
cuda Add GatherElements op and cuda kernel (#149) 2023-10-12 09:18:12 +08:00
intelcpu memory_allocator (#103) 2023-08-13 13:39:35 +08:00
kunlun Xpu task4 support: add softmax (#172) 2023-10-30 16:01:05 +08:00
nativecpu add transpose, concat and split for native cpu (#158) 2023-10-12 10:14:28 +08:00