TBE实战算子开发—DSL方式

前言

TBE算子详解

Hello World

• 目标：
  使用DSL语言实现平方根功能的TBE算子。
• 接口命名：
  sqrt_cce()。
• 算子分析：
  Sqrt算子功能是对Tensor中每个原子值求开方，数学表达式子为y=√𝒙 。
  根据当前TBE框架可支持的计算描述API，采用如下公式来表达Sqrt算子的计算过程：
  

TBE算子代码结构

• TBE算子入参

def sqrt_cce  (shape, dtype, kernel_name="sqrt_cce", need_build=False, need_print=False)