CANN/asc-devkit算子属性设置API
AddAttr【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit功能说明设置算子的属性以及对应值函数原型ContextBuilder AddAttr(const std::string attrName, int64_t attrValue) ContextBuilder AddAttr(const std::string attrName, bool attrValue) ContextBuilder AddAttr(const std::string attrName, const std::string attrValue) ContextBuilder AddAttr(const std::string attrName, float attrValue) ContextBuilder AddAttr(const std::string attrName, const std::vectorfloat attrValue) ContextBuilder AddAttr(const std::string attrName, const std::vectorbool attrValue) ContextBuilder AddAttr(const std::string attrName, const std::vectorint64_t attrValue) ContextBuilder AddAttr(const std::string attrName, const std::vectorstd::string attrValue) ContextBuilder AddAttr(const std::string attrName, const std::vectorstd::vectorint64_t attrValue)参数说明参数输入/输出说明attrName输入算子的属性名称attrValue输入算子属性值支持int64_t、bool、std::string、float、std::vectorfloat、std::vectorint64_t、std::vectorstd::string、std::vectorbool、std::vectorstd::vectorint64_t类型的指定。返回值说明当前ContextBuilder的对象。约束说明AddAttr不支持重复添加同名的属性调用示例context_ascendc::ContextBuilder builder; auto builder .NodeIoNum(1,1) .IrInstanceNum({1}) .AddAttr(attr_1, 1) .AddAttr(attr_2, true) .AddAttr(attr_3, stringValue) .AddAttr(attr_4, 1.f) .AddAttr(attr_5, {1}) .AddAttr(attr_6, {false}) .AddAttr(attr_7, {stringValue}) .AddAttr(attr_8, {1.f}) .AddAttr(attr_9, {{1, 2}, {3, 4}})【免费下载链接】asc-devkit本项目是CANN 推出的昇腾AI处理器专用的算子程序开发语言原生支持C和C标准规范主要由类库和语言扩展层构成提供多层级API满足多维场景算子开发诉求。项目地址: https://gitcode.com/cann/asc-devkit创作声明:本文部分内容由AI辅助生成(AIGC),仅供参考
本文来自互联网用户投稿,该文观点仅代表作者本人,不代表本站立场。本站仅提供信息存储空间服务,不拥有所有权,不承担相关法律责任。如若转载,请注明出处:http://www.coloradmin.cn/o/2599169.html
如若内容造成侵权/违法违规/事实不符,请联系多彩编程网进行投诉反馈,一经查实,立即删除!