已开启
[Requirement|需求建议]: add InferDataType for masked_fill, roll, slice, strided_slice, strided_slice_v2, strided_slice_v3, bias_add_grad, cdist, diag_part, invert, tile #2899
xufeng12121创建于  11 天前
xufeng12121
11 天前 创建

Thanks for sending an requirement! Please fill in the following template to help quickly solve your problem.

一、背景信息 (必填)

add InferDataType for masked_fill, roll, slice, strided_slice, strided_slice_v2, strided_slice_v3, bias_add_grad, cdist, diag_part, invert, tile

二、价值/作用 (必填)

masked_fill, roll, slice, strided_slice, strided_slice_v2,
strided_slice_v3, bias_add_grad, cdist, diag_part, invert, tile
等算子添加inferDataType函数

三、设计方案 (必填)

3.1 使能方式(涉及哪些框架:如Aclnn直调、Pytorch训练等)
3.2 总体设计

新增InferDataType 函数

3.3 支持硬件

A5

3.4 算子约束限制

💡 备注(选填)

likedislike
xufeng12121
11 天前 评论:

/assign

likedislike
CANN-robotCANN-robot成员
11 天前 将 xufeng12121 设为负责人