MobulaOP:简单灵活的跨框架操作员工具包 源码
MobulaOP Linux 视窗 覆盖范围 徽章 它是什么? MobulaOP是一个简单而灵活的跨框架操作员工具包。 您可以通过Python / C ++ / C / CUDA / HIP / TVM编写自定义运算符,而无需从源代码重建深度学习框架。 如何使用它? [] [] 添加加法运算符[] import mobula @ mobula . op . register class MyFirstOP : def forward ( self , x , y ): return x + y def backward ( self , dy ): return [ dy , dy ] def infer_shape ( self , in_shape ): assert in_shape [ 0 ]
文件列表
MobulaOP：简单灵活的跨框架操作员工具包
(预估有个131文件)
Convolution.cpp
5KB
ROIAlign.cpp
7KB
Transpose.cpp
781B
FocalLoss.cpp
2KB
Softmax.cpp
2KB
IoULoss.cpp
7KB
Sum.cpp
240B
async_mx_code.cpp
472B
kernel_code.cpp
167B
header.cpp
294B
暂无评论