InfiniTensor/include
bolun 7945693131 Add: benchmark for cudnn softmax 2023-07-20 11:03:07 +08:00
..
bang Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
core fix ReduceMean and element_wise (#90) 2023-06-29 07:15:07 +08:00
cuda Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
ffi Add TVM codegen for MemboundOp (#35) 2022-09-22 18:06:45 +08:00
intelcpu Cpu backend2 (#77) 2023-04-17 12:15:23 +08:00
nnet Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
operators Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
utils NNET supports TVM backend and kernels (#78) 2023-04-18 00:26:36 +08:00
benchmark.h Add: benchmark for cudnn softmax 2023-07-20 11:03:07 +08:00
test.h Add python interface for CUDA operator evaluation (#42) 2022-09-27 10:41:12 +08:00