InfiniTensor/include/operators
Liyan Zheng 537b3b4ea4 Add: Membound operator serialization 2023-04-18 21:53:48 +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
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 Dev for 202303ddl (#66) 2023-04-18 15:10:33 +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: Membound operator serialization 2023-04-18 21:53:48 +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 Cpu backend2 (#77) 2023-04-17 12:15:23 +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 Dev for 202303ddl (#66) 2023-04-18 15:10:33 +08:00
unary.h Fix: OpType and print device tensors 2023-04-18 20:28:08 +08:00