已关闭
[Requirement|需求建议]: 支持LeftShift算子TensorFlow框架插件 #2202
songkai111创建于 7月10日关闭于 7月10日
7月10日 添加了label:requirement
songkai111
7月10日 评论:
7月10日 评论:
/assign @songkai111


7月10日 将 songkai111 设为负责人
7月10日 关闭了 issue
7月10日 添加了label:resolved
/assign @songkai111


一、背景信息 (必填)
ops-math 仓库需要支持 LeftShift 算子的 TensorFlow 框架适配。当前 PR 新增 LeftShift 算子的 TensorFlow 插件(
math/left_shift/framework/left_shift_tf_plugin.cpp),通过REGISTER_CUSTOM_OP注册算子,完成 TensorFlow 框架下 LeftShift 算子的接入。二、价值/作用 (必填)
使 ops-math 仓库具备在 TensorFlow 框架下解析和调度 LeftShift 算子的能力,支撑 TensorFlow 原生
tf.bitwise.left_shift算子在昇腾平台上的运行。三、设计方案 (必填)
3.1 使能方式
TensorFlow 框架(代码依据:
FrameworkType(TENSORFLOW)),算子执行在 AI_CPU(代码依据:ImplyType(ImplyType::AI_CPU))。3.2 总体设计
通过
REGISTER_CUSTOM_OP("LeftShift")注册算子,使用AutoMappingByOpFn实现参数的自动映射解析,无需手动编写参数解析逻辑。3.2.1 算子支持的数据类型
见后续算子实现补充。
3.2.2 host侧设计
插件层使用
AutoMappingByOpFn自动映射参数,无需额外的 host 侧解析逻辑。3.2.3 kernel侧设计
见后续算子 kernel 实现补充。
3.3 支持硬件
AI_CPU(代码依据:
ImplyType(ImplyType::AI_CPU))。3.4 算子约束限制
见后续算子实现补充。
💡 备注
关联 PR:https://gitcode.com/cann/ops-math/pull/3957