InfiniTensor/include/operators
Liyan Zheng 51cc042f56 Add: nchw to nhwc conversion
Fix: conv parameter error in to_onnx
2023-04-24 02:29:53 +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
activation_backward.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
any.h Add: reduce in Any 2023-04-23 21:36:12 +08:00
batch_norm.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
concat.h Add documentation for operators. 2023-02-13 22:51:15 +08:00
conv.h Add: nchw to nhwc conversion 2023-04-24 02:29:53 +08:00
conv2dreduce.h fsrcnn 2023-04-23 20:56:19 +08:00
det.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
dropout.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
element_wise.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +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 Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
membound.h Add: hash match for membound kernels 2023-04-20 17:16:01 +08:00
pad.h feat: 前端支持 pad 及单元测试 2023-02-15 11:41:06 +08:00
pooling.h fix: remove inline keyword in class; rename getter and setter for inputOf and outputOf 2023-03-25 12:04:24 +08:00
reduce_mean.h feat: 导出 ReduceMean 到 onnx 2023-03-15 15:09:12 +08:00
reshape.h Add: Longformer models 2023-04-22 16:00:29 +08:00
resize.h Cpu backend2 (#77) 2023-04-17 12:15:23 +08:00
slice.h Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
softmax.h Cpu backend2 (#77) 2023-04-17 12:15:23 +08:00
split.h ADD: sub graph replacement. (#56) 2023-04-17 13:09:07 +08:00
transpose.h Add: Reshape/Transpose elimination 2023-04-23 02:10:05 +08:00
unary.h Fix: OpType and print device tensors 2023-04-18 20:28:08 +08:00