Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
A5 reduce模板算子infershape切换2.0
和A2/A3 infershape分开,新增noop_with_empty_axes属性支持
NA
新增infershape2.0公共方法
Ascend950
💡 备注(选填)
/assign @Sunee
Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.
一、背景信息 (必填)
A5 reduce模板算子infershape切换2.0
二、价值/作用 (必填)
和A2/A3 infershape分开,新增noop_with_empty_axes属性支持
三、设计方案 (必填)
3.1 使能方式(涉及哪些框架:如Aclnn直调、Pytorch训练等)
NA
3.2 总体设计
3.2.1 算子支持的数据类型
NA
3.2.2 host侧设计
新增infershape2.0公共方法
3.2.3 kernel侧设计
NA
3.3 支持硬件
Ascend950
3.4 算子约束限制
💡 备注(选填)