Closed
gather_nd算子支持Ascend950 #820
cpy_123456Add on  Feb 5Closed at  Feb 9
cpy_123456Collaborators
Feb 5 Create

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

Backgroud(背景信息)

gather_nd算子支持Ascend950(开源迁移)
算子功能:对于维度为r≥1的输入张量self,和维度q≥1的输入张量indices,将数据切片收集到维度为 (q-1) + (r - indices_shape[-1]) 的输出张量out中。
关联Pr:https://gitcode.com/cann/canndev/pull/80634

Origin(信息来源)

turing

Benefit / Necessity (价值/作用)

gather_nd算子支持Ascend950

Design(设计方案)

gather_nd算子支持Ascend950
输入输出设计

参数名 输入/输出 描述 数据类型 数据格式
self 输入 输入aclTensor。 INT64、INT32、INT8、UINT8、BOOL、FLOAT、FLOAT16、BFLOAT16、DOUBLE、INT16、UINT16、UINT32、UINT64 ND
indices 输入 输入aclTensor。 INT32、INT64 ND
negativeIndexSupport 输入 onnx模型是否存在负索引。 BOOL -
out 输出 输出aclTensor。 与self一致 ND
likedislike
Ccpy_123456Collaborators
Feb 5 Associated pull request
jiangjiaweiCollaborators
Feb 5 Comments:

/assign@cpy_123456

likedislike
cpy_123456Collaborators
Feb 5 Comments:

/assign

likedislike
CANN-robotCANN-robotCollaborators
Feb 5 Assign cpy_123456 as responsible person
Ccpy_123456Collaborators
Feb 6 Change title to "gather_nd算子支持Ascend950", originally titled "开源开放gather_nd算子".
Ccpy_123456Collaborators
Feb 6 Modify description of issue
Ccpy_123456Collaborators
Feb 6 Modify description of issue
Ccpy_123456Collaborators
Feb 6 Modify description of issue
Ccpy_123456Collaborators
Feb 6 Modify description of issue
CANN-robotCANN-robotCollaborators
Feb 9 Close issue
CANN-robotCANN-robotCollaborators
Mar 20 Added label: resolved