Edge算子样例

概述

样例概述：本样例介绍了利用ATVC实现自定义Edge单算子并完成功能验证

• 算子功能：自定义Edge计算的功能(一个元素的结果为周围相邻元素通过自定义计算得到的结果)
• 使用的ATVC模板：Pool
• 调用方式：Kernel直调

样例支持AI处理器型号：

• Ascend 910C
• Ascend 910B

算子描述

自定义Edge算子数学计算公式：

输入为二维数组：例如
x = [ x0, x1, x2, ...
      x3, x4, x5, ...
      x6, x7, x8, ...]

y4 = min(abs(((x2 + x5 + x8) - (x0 + x3 + x6)) / 3), 255)
以此类推其他元素的计算结果。

自定义Edge算子规格：

算子类型(OpType)	Edge
算子输入
	name	width	height	data type	format
	x	1023	2517	float	ND
算子输出
	z	1023	2517	float	ND
核函数名	EdgeCustom
规格限制说明	当前模板只支持2维shape按16元素个数对齐、 TILE_LAYOUT{16, 16}、TILE_PADDING{8, 8, 1, 1}的场景

目录结构

文件名	描述
edge.cpp	自定义Edge算子代码实现以及调用样例

算子运行

在ascendc-api-adv代码仓目录下执行：

cd ./ops_templates/atvc/examples
bash run_examples.sh edge

当前PoolOpTemplate暂不支持ATVC调试调优功能，相关功能待后续补充。