文档
注册

自定义算子模板

CANN工具提供了自定义算子模板,供用户学习和参考。

自定义算子模板所在路径为:CANN软件安装后文件存储路径中的“tools/msopgen/template/custom_operator_sample”。

custom_operator_sample目录结构。

├── AICPU
│   ├── Onnx
│   └── Tensorflow
├── DSL
│   ├── Mindspore
│   ├── Onnx
│   ├── PyTorch
│   └── Tensorflow
└── TIK
    ├── Mindspore
    ├── PyTorch
    └── Tensorflow
  • 自定义算子模板代码仅作为示例,供算子开发过程参考,不支持用户基于自定义算子模板进行产品化开发。
  • Onnx算子编译时需要protoc版本为3.13.0。
搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词