| .. |
|
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: reconfig ResizeObj, support "tf_crop_and_resize " and cubic coeff kernel. (#59)
|
2022-12-24 04:02:21 +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 |
d780f687fc