InfiniTensor/include
constroy 579cdbbb81
fix ReduceMean and element_wise (#90)
* feat: 导出 getPerfTime 到 python

Signed-off-by: YdrMaster <ydrml@hotmail.com>

* fix parsing of ReduceMean

* ReduceMean axes defaults to None

* fix ElementWiseCudnn with shape broadcasting

* fix format

---------

Signed-off-by: YdrMaster <ydrml@hotmail.com>
Co-authored-by: YdrMaster <ydrml@hotmail.com>
2023-06-29 07:15: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
test.h Add python interface for CUDA operator evaluation (#42) 2022-09-27 10:41:12 +08:00