build
函数功能
使用LLVM编译生成在CPU上运行的DSL算子。
函数原型
def build(inputs, args, name="default_function"):
参数说明
参数名 |
类型 |
说明 |
---|---|---|
inputs |
TVM的Schedule的对象 |
需要编译的TVM的调度实例对象。 |
args |
Tensor列表 |
算子的输入、输出参数列表,类型为tensor。 |
name |
字符串 |
编译生成的算子二进制文件名称。 默认值为“default_function”。 |
返回值
无
约束说明
无
父主题: TBE DSL Testing API