InfiniTensor/include/operators
zhangyunze ef672894d0
support mixed dtype (#102)
* feat: support mixed dtype

* feat: support cast op

* test: add test for cast op

* feat: support datatype BFloat16

* feat: support data convert fp32 <-> bfp16

* fix: fix all op's infershape func

* fix as review comment
2023-08-16 21:49:43 +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 refactor(core): 添加新的 `OpType` 定义 (#99) 2023-08-07 11:17:05 +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 NNET supports TVM backend and kernels (#78) 2023-04-18 00:26:36 +08:00
pad.h feat: 前端支持 pad 及单元测试 2023-02-15 11:41:06 +08:00
pooling.h refactor(core): 添加新的 `OpType` 定义 (#99) 2023-08-07 11:17:05 +08:00
reduce_mean.h feat: 导出 ReduceMean 到 onnx 2023-03-15 15:09:12 +08:00
reshape.h 支持fp16 dtype (#96) 2023-08-02 16:38:16 +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 support mixed dtype (#102) 2023-08-16 21:49:43 +08:00
unary.h support mixed dtype (#102) 2023-08-16 21:49:43 +08:00