InfiniTensor/include
wanghailu 6b53a50927 add hardtanh operation 2023-01-29 02:37:15 +00:00
..
bang add operation cnnl div, test and test for divdemo bangc kernel 2022-12-15 08:26:49 +00:00
core add hardtanh operation 2023-01-29 02:37:15 +00:00
cuda Add: resize operator and cuda kernel,support nearest/linear coef. (#51) 2022-11-14 09:30:22 +08:00
ffi Add TVM codegen for MemboundOp (#35) 2022-09-22 18:06:45 +08:00
nnet Fix NNet tests after migration (#27) 2022-09-13 15:17:22 +08:00
operators add hardtanh operation 2023-01-29 02:37:15 +00:00
utils ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
test.h Add python interface for CUDA operator evaluation (#42) 2022-09-27 10:41:12 +08:00