样例使用
功能描述
此样例通过TIK方式实现了BatchNorm算子,并支持动态shape场景下的算子执行。
BatchNorm算子是对输入进行归一化处理,xnorm=(x−μ)/σ,其中, μ和 σ是计算的均值和方差。
样例获取
单击Gitee或Github,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“samples/cplusplus/level1_single_api/4_op_dev/4_other/singleop/tbe/impl”目录中获取TIK动态shape算子的实现样例。
单击Gitee或Github,进入Ascend samples开源仓,参见README中的“版本说明”下载配套版本的sample包,从“samples/cplusplus/level1_single_api/4_op_dev/2_verify_op/acl_execute_batchnorm”目录中获取TIK动态shape算子的验证样例。
算子编译
参考“samples/cplusplus/level1_single_api/4_op_dev/4_other/singleop/tbe/impl”目录的readme进行算子编译。
Ascend EP场景,在Host侧运行样例
参考“samples/cplusplus/level1_single_api/4_op_dev/2_verify_op/acl_execute_batchnorm”目录的readme编译BatchNorm算子运行工程,生成单算子验证可执行文件,并运行该可执行文件。
父主题: TIK自定义算子动态Shape专题