参考实现
本章以BatchNorm算子为例,介绍如何通过TIK方式开发自定义算子,并在运行时支持动态shape。
设计思路
算子实现
算子选择器
算子调用
父主题:
TIK自定义算子动态Shape专题