07bd16b8创建于 4月17日历史提交

文件	最后提交记录	最后更新时间
docs	doc tools低错内容4.17 Co-authored-by: caiwenwen<caiwenwen6@h-partners.com> # message auto-generated for no-merge-commit merge: !3912 merge 9.0.0 into 9.0.0 doc tools低错内容4.17 Created-by: caiwenwen Commit-by: caiwenwen Merged-by: cann-robot Description: ## 描述修改markdown语法300+ ## 关联的Issue 关联Issue [#1783](https://gitcode.com/cann/ops-nn/issues/1783) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新全部文件 ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [X] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!3912	4 个月前
examples	代码同步 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !450 merge master into master 代码同步 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!450	8 个月前
op_graph	代码同步 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !450 merge master into master 代码同步 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> <!-- 如果这个PR是为了解决特定的问题单，请在这里描述问题单单号。--> ## 测试 <!--描述进行了哪些测试来验证你的改动。包括但不限于二级冒烟、算子泛化等。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!450	8 个月前
op_host	整数attr获取风险整改 Co-authored-by: zhangqijia1<zhangqijia1@h-partners.com> # message auto-generated for no-merge-commit merge: !2725 merge fix_get_point_9.0.0 into 9.0.0 整数attr获取风险整改 Created-by: gcw_SUaZx3UQ Commit-by: zhangqijia1 Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 部分算子的infershape和tiling中获取算子整型的属性用int类型或者int32_t类型，相关接口的属性使用int64_t存储，用int获取，存在数据截断的风险。需要整改。 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> [#1507](https://gitcode.com/cann/ops-nn/issues/1507) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2725	5 个月前
op_kernel	修改maxPool3DWithArgmaxV2、maxPoolV2、maxPool3DGradWithArgmax、avgPool3DGrad、embeddingBag、embeddingDenseGradV2编译告警 Co-authored-by: SimonZzz<zhouquan79@huawei.com> # message auto-generated for no-merge-commit merge: !2013 merge fixcl into master 修改maxPool3DWithArgmaxV2、maxPoolV2、maxPool3DGradWithArgmax、avgPool3DGrad、embeddingBag、embeddingDenseGradV2编译告警 Created-by: SimonZzz Commit-by: SimonZzz Merged-by: cann-robot Description: ## 描述 <!--在这里详细描述你的改动，包括改动的原因和所采取的方法。--> 修改maxPool3DWithArgmaxV2、maxPoolV2、maxPool3DGradWithArgmax、avgPool3DGrad、embeddingBag、embeddingDenseGradV2编译告警 ## 关联的Issue <!-- 如果这个PR是为了解决特定的Issue，请在这里提供Issue链接。例如：关联Issue #000--> https://gitcode.com/cann/ops-nn/issues/1120 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> 冒烟已过 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2013	5 个月前
tests	[aicore]整改 Co-authored-by: yang-di52<yangdi52@huawei.com> # message auto-generated for no-merge-commit merge: !2261 merge master into master [aicore]整改 Created-by: yang-di52 Commit-by: yang-di52 Merged-by: cann-robot Description: ## 描述编译器内部已经定义了__aicore__，算子侧需要去掉这个标识 ## 关联的Issue [#1276](https://gitcode.com/cann/ops-nn/issues/1276) ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!2261	5 个月前
CMakeLists.txt	License Change Split2 Co-authored-by: huohuo_wy<wangyan389@huawei.com> # message auto-generated for no-merge-commit merge: !371 merge spilit2License3 into master License Change Split2 Created-by: huohuo_wangyan Commit-by: huohuo_wy Merged-by: cann-robot Description: ## 描述批量整改文件copyright注释 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/177 ## 测试 COPYRIGHT注释头按新要求刷新 ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [x] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!371	8 个月前
README.md	修改算子readme文档 Co-authored-by: xiu_ling_wang<wangxiuling2@h-partners.com> # message auto-generated for no-merge-commit merge: !3615 merge md-modify-9.0.0 into 9.0.0 修改算子readme文档 Created-by: xiu_ling_wang Commit-by: xiu_ling_wang Merged-by: cann-robot Description: ## 描述修改算子readme文档。 ## 关联的Issue https://gitcode.com/cann/ops-nn/issues/1772 ## 测试 <!--描述进行了哪些测试来验证你的改动。--> ## 文档更新 <!--如果这个PR包含文档的更新，请在这里指出。例如：更新了README.md文件。--> ## 类型标签 <!-- [x] 表示选中 --> - [ ] Bug修复 - [ ] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-nn!3615	4 个月前

EmbeddingBag

产品支持情况

产品	是否支持
Ascend 950PR/Ascend 950DT	√
Atlas A3 训练系列产品/Atlas A3 推理系列产品	√
Atlas A2 训练系列产品/Atlas A2 推理系列产品	√
Atlas 200I/500 A2推理产品	×
Atlas 推理系列产品	√
Atlas 训练系列产品	√
Kirin X90 处理器系列产品	√
Kirin 9030 处理器系列产品	√

功能说明

算子功能：根据indices从weight中获得一组被聚合的数，然后根据offsets的偏移和mode指定的聚合模式对获取的数进行max、sum、mean聚合。其余参数则更细化了计算过程的控制。

shape推导方式如下：假设:

weight的shape为(numWeight, embeddingDim)
indices的shape为(indices)
offsets的shape为(offsets)

当mode为sum模式：

y的shape 为 include_last_offset ? (offsets - 1, embeddingDim) : (offsets, embeddingDim)
offset2bag的shape 为 (indices,)
bag_size的shape 为 include_last_offset ? (offsets - 1) : (offsets,)
max_indices的shape 为 include_last_offset ? (offsets - 1) : (offsets,)

当mode为mean模式：

y的shape 为 include_last_offset? (offsets - 1, embeddingDim) : (offsets, embeddingDim)
offset2bag的shape 为 (indices,)
bag_size的shape 为 include_last_offset ? (offsets - 1) : (offsets,)
max_indices的shape 为 include_last_offset ? (offsets - 1) : (offsets,)

当mode为max模式：

y的shape 为 include_last_offset ? (offsets - 1, embeddingDim) : (offsets, embeddingDim)
offset2bag的shape 为 (indices,)
bag_size的shape 为 include_last_offset ? (offsets - 1) : (offsets,)
max_indices的shape 为 include_last_offset ? (offsets - 1, embeddingDim) : (offsets, embeddingDim)

参数说明

参数名	输入/输出/属性	描述	数据类型	数据格式
weight	输入	词嵌入矩阵，包含所有词的嵌入向量。	FLOAT、FLOAT16、BFLOAT16	ND
indices	输入	包含索引的张量，指定要从weight中提取哪些词的嵌入向量。	INT32、INT64	ND
offsets	可选输入	用于将indices分割成多个bag的偏移量张量。	INT32、INT64	ND
per_sample_weights	可选输入	指定样本权重。	FLOAT、FLOAT16、BFLOAT16	ND
y	输出	词嵌入矩阵聚合后的结果。	FLOAT、FLOAT16、BFLOAT16	ND
offset2bag	输出	bag的起始偏移。	INT32、INT64	ND
bag_size	输出	每个bag的大小。	INT32、INT64	ND
max_indices	输出	当mode为max时，词嵌入向量最大值所在的行。	INT32、INT64	ND
mode	可选属性	用于控制聚合模式。默认值为"mean"。	String	-
scale_grad_by_freq	可选属性	用于控制是否根据词频缩放梯度。默认值为false。	Bool	-
sparse	可选属性	用于控制稀疏模式。默认值为false。	Bool	-
include_last_offset	可选属性	控制是否包含最后的偏移。默认值为false。	Bool	-
padding_idx	可选属性	控制是否包含最后的偏移。默认值为-1。	Int	-

Atlas 训练系列产品：不支持BFLOAT16。
Kirin X90/Kirin 9030处理器系列产品：不支持BFLOAT16。

约束说明

无

调用说明

调用方式	调用样例	说明
aclnn调用	test_aclnn_embedding_bag	通过aclnnEmbeddingBag接口方式调用EmbeddingBag算子。
图模式调用	test_geir_embedding_bag	通过算子IR构图方式调用EmbeddingBag算子。