InfiniTensor/include/core
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
..
blob.h Support bang c kernel wanghailu 0927 (#43) 2022-09-30 11:01:52 +08:00
common.h Add ConvTransposedNHWC (#67) 2023-03-01 14:15:02 +08:00
constants.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
data_type.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
dummy_mutator.h Add search engine (#64) 2023-02-12 18:27:52 +08:00
graph.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
graph_handler.h fix ReduceMean and element_wise (#90) 2023-06-29 07:15:07 +08:00
graph_match.h ADD: sub graph replacement. (#56) 2023-04-17 13:09:07 +08:00
hash.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
kernel.h Add search engine (#64) 2023-02-12 18:27:52 +08:00
mutator.h ADD: add mkl runtime for intel cpu , and add mkl kernel for matmul/conv/convtransposed. (#61) 2023-03-27 21:28:49 +08:00
object.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
operator.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
perf_engine.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
ref.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
runtime.h ADD: sub graph replacement. (#56) 2023-04-17 13:09:07 +08:00
search_engine.h Add search engine (#64) 2023-02-12 18:27:52 +08:00
tensor.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
tensor_base.h ADD: sub graph replacement. (#56) 2023-04-17 13:09:07 +08:00