InfiniTensor/include/operators
wendy12022 c5966f8d81
Add: resize operator and cuda kernel,support nearest/linear coef. (#51)
ADD: resize operator and cuda kernel,support nearest/linear coef.

fix some

fix tests

add more tests for linear mode.

add linear coef mode.

add scales

add tests

fix tests.

add notLarger notSmaller

fix

add test

ADD:resize operator and cuda kernel
2022-11-14 09:30:22 +08:00
..
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