InfiniTensor/src/operators
YdrMaster 4ffaa44c1e fix: Matmul 支持 2 维或以上的输入
> 现在能导入 resnet18

Signed-off-by: YdrMaster <ydrml@hotmail.com>
2023-03-15 17:23:32 +08:00
..
G2BMM.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
GBMM.cc Fix CMake USE_CUDA (#36) 2022-09-21 12:28:00 +08:00
batch_norm.cc fix: 修正 batchNorm 实现 2023-03-15 17:23:32 +08:00
concat.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
conv.cc Add ConvTransposedNHWC (#67) 2023-03-01 14:15:02 +08:00
element_wise.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
extend.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
gather.cc feat: 前端支持 gather 及单元测试 2023-02-14 14:16:01 +08:00
matmul.cc fix: Matmul 支持 2 维或以上的输入 2023-03-15 17:23:32 +08:00
membound.cc ADD: batch norm operator and cuda kernel. (#44) 2022-10-15 16:29:28 +08:00
pad.cc feat: 前端支持 pad 及单元测试 2023-02-15 11:41:06 +08:00
pooling.cc feat: 前端支持 pool 及单元测试 2023-02-14 16:26:47 +08:00
reduce_mean.cc fix: 移除 c++ 中的中文注释,python TODO 改 FIXME 2023-03-15 14:48:39 +08:00
reshape.cc feat: 前端支持 reshape 2023-02-14 09:51:11 +08:00
resize.cc ADD: reconfig ResizeObj, support "tf_crop_and_resize " and cubic coeff kernel. (#59) 2022-12-24 04:02:21 +08:00
slice.cc opt: 优化 SliceObj 构造器实现 2023-02-14 16:44:08 +08:00
split.cc opt: 优化 PadObj 和 SplitObj 构造器实现 2023-02-15 11:28:49 +08:00
unary.cc Add activation operators and kernels 2022-09-16 13:58:57 +08:00