已关闭
[Bug-Report|缺陷反馈]: atan2算子缺少infershape和inferdtype注册 #2613
clinglai0517创建于  8月13日关闭于  8月13日
clinglai0517成员
8月13日 创建

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

一、问题描述 (必填)

正式版 math/atan2 算子未注册 GE 图层的 InferShape 和 InferDataType:

  • op_host 目录下没有 atan2_infershape.cpp / inferdtype 相关实现;
  • op_graph/atan2_proto.h 中 REG_OP(Atan2) 仅声明了 INPUT/OUTPUT 和 TensorType,未注册 INFER_FUNC/INFER_DTYPE;
  • op_host/atan2_def.cpp 只注册了 OpDef 的 dtype/format 支持列表,未注册任何推断函数。

影响

  • 图编译/静态 shape 推导场景下,Atan2 缺少输出 shape 和 dtype 的显式推断,无法保证广播语义下输出 shape 正确推导;
  • 与 math 目录下同类算子(atan、acos、add_v2、zeta 等均注册 IMPL_OP_INFERSHAPE)不一致。

二、环境信息 (可选)

三、重现步骤 (可选)

从图模式调用atan2必现

四、预期结果 (可选)

在图模式下能正常推导输出shape和dtype

💡 备注(选填)

likedislike
陈思
陈思成员
8月13日 评论:

/assign @clinglai0517

likedislike
CANN-robotCANN-robot成员
8月13日 将 clinglai0517 设为负责人
CANN-robotCANN-robot成员
8月13日 关闭了 issue
CANN-robotCANN-robot成员
8月13日 添加了label:resolved