InfiniTensor/include/operators
YdrMaster a5e692baea feat: 导出 batchnorm 到 onnx
Signed-off-by: YdrMaster <ydrml@hotmail.com>
2023-03-15 17:23:32 +08:00
..
G2BMM.h Add documentation for operators. 2023-02-13 22:51:15 +08:00
GBMM.h Add documentation for operators. 2023-02-13 22:51:15 +08:00
batch_norm.h feat: 导出 batchnorm 到 onnx 2023-03-15 17:23:32 +08:00
concat.h Add documentation for operators. 2023-02-13 22:51:15 +08:00
conv.h Merge branch 'master' into dev-onnx 2023-03-15 14:52:03 +08:00
element_wise.h Add documentation for operators. 2023-02-13 22:51:15 +08:00
extend.h Add documentation for operators. 2023-02-13 22:51:15 +08:00
gather.h feat: 前端支持 gather 及单元测试 2023-02-14 14:16:01 +08:00
matmul.h Add documentation for operators. 2023-02-13 22:51:15 +08:00
membound.h Add search engine (#64) 2023-02-12 18:27:52 +08:00
pad.h feat: 前端支持 pad 及单元测试 2023-02-15 11:41:06 +08:00
pooling.h Add documentation for operators. 2023-02-13 22:51:15 +08:00
reduce_mean.h feat: 导出 ReduceMean 到 onnx 2023-03-15 15:09:12 +08:00
reshape.h feat: 导出 Reshape 到 onnx 2023-03-15 15:09:12 +08:00
resize.h Add documentation for operators. 2023-02-13 22:51:15 +08:00
slice.h opt: 优化 SliceObj 构造器实现 2023-02-14 16:44:08 +08:00
split.h Add documentation for operators. 2023-02-13 22:51:15 +08:00
unary.h Add documentation for operators. 2023-02-13 22:51:15 +08:00