已关闭
[Requirement|需求建议]: 950新增StridedSliceAssign算子 #2971
xuejinghui创建于 8 天前关闭于 8 天前
8 天前 添加了label:requirement
8 天前 修改了issue 的描述
Xxuejinghui
8 天前 修改标题为 “[Requirement|需求建议]: 950新增StridedSliceAssign算子”,原标题为“[Requirement|需求建议]: 950新增Col2ImV2算子”
8 天前 修改标题为 “[Requirement|需求建议]: 950新增StridedSliceAssign算子”,原标题为“[Requirement|需求建议]: 950新增Col2ImV2算子”
xuejinghui
8 天前 评论:
8 天前 评论:
/assign


8 天前 将 xuejinghui 设为负责人
8 天前 关闭了 issue
8 天前 添加了label:resolved
一、背景信息 (必填)
在 Ascend 950(950PR/950DT)上新增 StridedSliceAssign 算子(非 D 版)。
功能:将输入张量 input_value 的内容,赋值给目标张量 var 中由 begin、end、strides 三个 1 维 int64 常量输入(配合 begin_mask/end_mask/ellipsis_mask/new_axis_mask/shrink_axis_mask 五个属性)指定的带步长切片位置,切片以外区域保持 var 原值不变(in-place / ref 语义,输出 var 与输入 var 同 shape 同 dtype)。
语义对齐 TensorFlow
tf.raw_ops.StridedSliceAssign(稀疏切片 spec 形式:begin/end/strides 为 tensor 输入而非属性),补齐 950 平台该 TF 兼容算子的缺口,解决 TF 图迁移场景下该算子在 950 上无可调用实现的问题。二、价值/作用 (必填)
三、设计方案 (必填)
3.1 使能方式(涉及哪些框架:如Aclnn直调、Pytorch训练等)
op_graph/strided_slice_assign_proto.h内注册(带OPS_PROTO_DEF_STRIDEDSLICEASSIGN宏隔离),TF 框架经适配器构图接入3.2 总体设计
3.2.1 算子支持的数据类型
属性:begin_mask / end_mask / ellipsis_mask / new_axis_mask / shrink_axis_mask(int,默认 0)
3.2.2 host侧设计
OP_LOGE_FOR_INVALID_*宏3.2.3 kernel侧设计
3.3 支持硬件
3.4 算子约束限制
💡 备注(选填)
tf.raw_ops.StridedSliceAssign(ref 语义经tf.compat.v1.Variable(use_resource=False)承载);TTK 黑盒 150 + 白盒 198 + 网络 13 全量通过(361/361)const2attr_fusion_pass.cc中REGISTER_CONST2ATTR("StridedSliceAssignD").OriginOpType("StridedSliceAssign")