InfiniTensor/include
zhengly123 4e0040c8a0
Add: connection among tensors and operators (#45)
* Add: refs_to_wrefs and wrefs_to_refs

* Add: op and tensor connection

* Add: inception-v3 block test

* Refactor: addOperatorAndConnect

Co-authored-by: Liyan Zheng <liyan-zheng@outlook.com>
2022-10-18 22:02:51 +08:00
..
bang Support bang c kernel wanghailu 0927 (#43) 2022-09-30 11:01:52 +08:00
core Add: connection among tensors and operators (#45) 2022-10-18 22:02:51 +08:00
cuda ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +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:reduce_mean operator and cuda kernel. (#47) 2022-10-15 16:53:58 +08: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