..
G2BMM.h
ADD: batch norm operator and cuda kernel. ( #44 )
2022-10-15 16:29:28 +08:00
GBMM.h
ADD: batch norm operator and cuda kernel. ( #44 )
2022-10-15 16:29:28 +08:00
batch_norm.h
ADD: batch norm operator and cuda kernel. ( #44 )
2022-10-15 16:29:28 +08:00
concat.h
ADD: batch norm operator and cuda kernel. ( #44 )
2022-10-15 16:29:28 +08:00
conv.h
Add: ConvTransposed ( #33 )
2022-09-19 15:05:39 +08:00
element_wise.h
ADD: batch norm operator and cuda kernel. ( #44 )
2022-10-15 16:29:28 +08:00
extend.h
ADD: batch norm operator and cuda kernel. ( #44 )
2022-10-15 16:29:28 +08:00
gather.h
ADD: batch norm operator and cuda kernel. ( #44 )
2022-10-15 16:29:28 +08:00
matmul.h
Fix NNet tests after migration ( #27 )
2022-09-13 15:17:22 +08:00
membound.h
Fix NNet tests after migration ( #27 )
2022-09-13 15:17:22 +08:00
pad.h
ADD: batch norm operator and cuda kernel. ( #44 )
2022-10-15 16:29:28 +08:00
pooling.h
ADD: batch norm operator and cuda kernel. ( #44 )
2022-10-15 16:29:28 +08:00
reduce_mean.h
ADD:reduce_mean operator and cuda kernel. ( #47 )
2022-10-15 16:53:58 +08:00
reshape.h
Add: reshape/flatten/identity OP and cuda kernel ( #34 )
2022-09-21 14:04:30 +08:00
resize.h
Add: resize operator and cuda kernel,support nearest/linear coef. ( #51 )
2022-11-14 09:30:22 +08:00
slice.h
ADD:pad/slice operator and cuda kernel. ( #39 )
2022-09-29 10:29:24 +08:00
split.h
ADD:concat/split operator and cuda kernels ( #29 )
2022-09-29 11:01:30 +08:00
unary.h
Add activation operators and kernels
2022-09-16 13:58:57 +08:00