代码拉取完成,页面将自动刷新
.. toctree:: :maxdepth: 1 :hidden: operation/op_custom_prim operation/op_custom_ascendc operation/op_custom_aot operation/op_custom_adv operation/op_customopbuilder operation/cpp_api_for_custom_ops operation/op_customopbuilder_atb operation/op_customopbuilder_asdsip operation/op_customopbuilder_aclnn
当开发网络遇到内置算子不足以满足需求时,你可以利用MindSpore的自定义算子功能接入你的算子。当前MindSpore提供了两种方式接入自定义算子,分别是 基于Custom原语接入 和 基于CustomOpBuilder接入 。
接口比较 | Custom原语 | CustomOpBuilder |
---|---|---|
支持模式 | 静态图(Graph Mode)和动态图(PyNative Mode) | 动态图(PyNative Mode) |
接口功能 | 提供统一的Custom原语,在算子执行的各个阶段分别调用用户接口。 | 在线编译和加载自定义算子模块,可以直接应用到网络当中。 |
优点 | 同时支持动态图和静态图,算子调度执行流程与内置算子一致,执行性能高效。 | 基于C++ tensor开发算子,自定义执行流程,更加直观。 |
缺点 | 接口限制较多,算子执行流程对用户不可见。 | 开发算子涉及的接口较多,当前暂无简洁高效的C++ API,开发高效算子的难度较大。 |
特性等级 | STABLE | BETA |
此处可能存在不合适展示的内容,页面不予展示。您可通过相关编辑功能自查并修改。
如您确认内容无涉及 不当用语 / 纯广告导流 / 暴力 / 低俗色情 / 侵权 / 盗版 / 虚假 / 无价值内容或违法国家有关法律法规的内容,可点击提交进行申诉,我们将尽快为您处理。