..
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
all_gather.h
impl distributed launch with NCCL ( #106 )
2023-09-05 09:47:35 +08:00
all_reduce.h
impl distributed launch with NCCL ( #106 )
2023-09-05 09:47:35 +08:00
batch_norm.h
Dev for 202303ddl ( #66 )
2023-04-18 15:10:33 +08:00
broadcast.h
impl distributed launch with NCCL ( #106 )
2023-09-05 09:47:35 +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
expand.h
框架支持bert/gpt2模型构图 ( #94 )
2023-08-29 16:06:52 +08:00
extend.h
Add documentation for operators.
2023-02-13 22:51:15 +08:00
gather.h
Add GatherElements op and cuda kernel ( #149 )
2023-10-12 09:18:12 +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
Pooling ceil mode ( #155 )
2023-10-09 20:51:39 +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
Add HardSigmoid and HardSwish ( #156 )
2023-10-10 22:41:06 +08:00
where.h
框架支持bert/gpt2模型构图 ( #94 )
2023-08-29 16:06:52 +08:00
36ae7b7fb6