InfiniTensor/test/kernels
xgqdut2016 a7293c12ba
Add layer normalization (#181)
* - add layernorm kernel

* success:add layernorm kernel and test

* fix: remove unusalble comments

* fix: modify code as reviewer suggested

* debug,modified .cu and test

* optional bias support

* overloading function

* fix bug after merging; remove time constrain in conv test

---------

Co-authored-by: kilinchange <kilinchange@163.com>
Co-authored-by: Haojie Wang <haojie0429@gmail.com>
2023-11-24 15:15:14 +08:00
..
bang Pooling ceil mode (#155) 2023-10-09 20:51:39 +08:00
cuda Add layer normalization (#181) 2023-11-24 15:15:14 +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