0
代码
代码
Issues
Pull Requests
讨论
Wiki
分析
0
cann-robotcann-robotadd aicpu:coordinates_1d_to_2d, ones_like
53224427创建于 13 天前历史提交
文件最后提交记录最后更新时间
examples
add aicpu:coordinates_1d_to_2d, ones_like Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !2785 merge dev_extra into master add aicpu:coordinates_1d_to_2d, ones_like Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 新增三个 AICPU 算子实现： - **coordinates_1d_to_2d**: 将1D坐标转换为2D坐标，根据shape信息计算行列索引 - **ones_like**: 返回形状和类型相同的张量，所有元素都设置为1 ## 关联的Issue #1584 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 执行 cmake 配置和编译，验证新增算子能正确构建 - 验证 CMakeLists.txt 配置正确 2. **算子加载测试** - 验证算子接口注册成功 - 测试 GE IR 模式加载 3. **单元测试** - 运行 tests/ut/op_kernel_aicpu 目录下的单元测试 - 验证算子计算逻辑正确性 4. **功能验证** - 验证 coordinates_1d_to_2d 坐标转换功能 - 验证 ones_like 张量生成功能 ## 文档更新 - 新增 conversion/coordinates_1d_to_2d/README.md - 新增算子功能说明和数据类型支持列表 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!278513 天前
op_graph
add aicpu:coordinates_1d_to_2d, ones_like Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !2785 merge dev_extra into master add aicpu:coordinates_1d_to_2d, ones_like Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 新增三个 AICPU 算子实现： - **coordinates_1d_to_2d**: 将1D坐标转换为2D坐标，根据shape信息计算行列索引 - **ones_like**: 返回形状和类型相同的张量，所有元素都设置为1 ## 关联的Issue #1584 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 执行 cmake 配置和编译，验证新增算子能正确构建 - 验证 CMakeLists.txt 配置正确 2. **算子加载测试** - 验证算子接口注册成功 - 测试 GE IR 模式加载 3. **单元测试** - 运行 tests/ut/op_kernel_aicpu 目录下的单元测试 - 验证算子计算逻辑正确性 4. **功能验证** - 验证 coordinates_1d_to_2d 坐标转换功能 - 验证 ones_like 张量生成功能 ## 文档更新 - 新增 conversion/coordinates_1d_to_2d/README.md - 新增算子功能说明和数据类型支持列表 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!278513 天前
op_kernel_aicpu
add aicpu:coordinates_1d_to_2d, ones_like Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !2785 merge dev_extra into master add aicpu:coordinates_1d_to_2d, ones_like Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 新增三个 AICPU 算子实现： - **coordinates_1d_to_2d**: 将1D坐标转换为2D坐标，根据shape信息计算行列索引 - **ones_like**: 返回形状和类型相同的张量，所有元素都设置为1 ## 关联的Issue #1584 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 执行 cmake 配置和编译，验证新增算子能正确构建 - 验证 CMakeLists.txt 配置正确 2. **算子加载测试** - 验证算子接口注册成功 - 测试 GE IR 模式加载 3. **单元测试** - 运行 tests/ut/op_kernel_aicpu 目录下的单元测试 - 验证算子计算逻辑正确性 4. **功能验证** - 验证 coordinates_1d_to_2d 坐标转换功能 - 验证 ones_like 张量生成功能 ## 文档更新 - 新增 conversion/coordinates_1d_to_2d/README.md - 新增算子功能说明和数据类型支持列表 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!278513 天前
tests
add aicpu:coordinates_1d_to_2d, ones_like Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !2785 merge dev_extra into master add aicpu:coordinates_1d_to_2d, ones_like Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 新增三个 AICPU 算子实现： - **coordinates_1d_to_2d**: 将1D坐标转换为2D坐标，根据shape信息计算行列索引 - **ones_like**: 返回形状和类型相同的张量，所有元素都设置为1 ## 关联的Issue #1584 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 执行 cmake 配置和编译，验证新增算子能正确构建 - 验证 CMakeLists.txt 配置正确 2. **算子加载测试** - 验证算子接口注册成功 - 测试 GE IR 模式加载 3. **单元测试** - 运行 tests/ut/op_kernel_aicpu 目录下的单元测试 - 验证算子计算逻辑正确性 4. **功能验证** - 验证 coordinates_1d_to_2d 坐标转换功能 - 验证 ones_like 张量生成功能 ## 文档更新 - 新增 conversion/coordinates_1d_to_2d/README.md - 新增算子功能说明和数据类型支持列表 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!278513 天前
CMakeLists.txt
add aicpu:coordinates_1d_to_2d, ones_like Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !2785 merge dev_extra into master add aicpu:coordinates_1d_to_2d, ones_like Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 新增三个 AICPU 算子实现： - **coordinates_1d_to_2d**: 将1D坐标转换为2D坐标，根据shape信息计算行列索引 - **ones_like**: 返回形状和类型相同的张量，所有元素都设置为1 ## 关联的Issue #1584 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 执行 cmake 配置和编译，验证新增算子能正确构建 - 验证 CMakeLists.txt 配置正确 2. **算子加载测试** - 验证算子接口注册成功 - 测试 GE IR 模式加载 3. **单元测试** - 运行 tests/ut/op_kernel_aicpu 目录下的单元测试 - 验证算子计算逻辑正确性 4. **功能验证** - 验证 coordinates_1d_to_2d 坐标转换功能 - 验证 ones_like 张量生成功能 ## 文档更新 - 新增 conversion/coordinates_1d_to_2d/README.md - 新增算子功能说明和数据类型支持列表 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!278513 天前
README.md
add aicpu:coordinates_1d_to_2d, ones_like Co-authored-by: sujunwei3<sujunwei3@huawei.com> # message auto-generated for no-merge-commit merge: !2785 merge dev_extra into master add aicpu:coordinates_1d_to_2d, ones_like Created-by: sujunwei3 Commit-by: sujunwei3 Merged-by: cann-robot Description: ## 描述 新增三个 AICPU 算子实现： - **coordinates_1d_to_2d**: 将1D坐标转换为2D坐标，根据shape信息计算行列索引 - **ones_like**: 返回形状和类型相同的张量，所有元素都设置为1 ## 关联的Issue #1584 ## 测试 根据代码变更，测试场景如下： 1. **编译构建测试** - 执行 cmake 配置和编译，验证新增算子能正确构建 - 验证 CMakeLists.txt 配置正确 2. **算子加载测试** - 验证算子接口注册成功 - 测试 GE IR 模式加载 3. **单元测试** - 运行 tests/ut/op_kernel_aicpu 目录下的单元测试 - 验证算子计算逻辑正确性 4. **功能验证** - 验证 coordinates_1d_to_2d 坐标转换功能 - 验证 ones_like 张量生成功能 ## 文档更新 - 新增 conversion/coordinates_1d_to_2d/README.md - 新增算子功能说明和数据类型支持列表 ## 类型标签 - [ ] Bug修复 - [x] 新特性 - [ ] 性能优化 - [ ] 文档更新 - [ ] 其他，请描述： See merge request: cann/ops-math!278513 天前
README.md

Coordinates1DTo2D

产品支持情况

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

功能说明

  • 算子功能：将1D坐标转换为2D坐标，根据shape信息计算行列索引。

参数说明

参数名 输入/输出/属性 描述 数据类型 数据格式
x 输入 1D坐标索引值。 INT32、INT64、UINT64 ND
shape 输入 形状信息，包含4个元素(N, D, H, W)。 INT32、INT64、UINT64 ND
output_row 输出 输出的行索引。 INT32、INT64、UINT64 ND
output_col 输出 输出的列索引。 INT32、INT64、UINT64 ND
output_n 输出 输出的列数(W维度值)。 INT32、INT64、UINT64 ND

约束说明

  • input[shape]元素个数必须为4
  • input[x]与input[shape]数据类型必须一致
  • shape[3]（W维度）不能为0

调用说明

调用方式 样例代码 说明
图模式接口 test_geir_coordinates_1d_to_2d 通过算子IR接口方式调用Coordinates1DTo2D算子。