文档
注册
评分
提单
论坛
小AI

build

函数功能

使用LLVM编译生成在CPU上运行的DSL算子。

函数原型

def build(inputs, args, name="default_function"):

参数说明

参数名

类型

说明

inputs

TVM的Schedule的对象

需要编译的TVM的调度实例对象。

args

Tensor列表

算子的输入、输出参数列表,类型为tensor。

name

字符串

编译生成的算子二进制文件名称。

默认值为“default_function”。

返回值

约束说明

调用示例

请参见调试示例

搜索结果
找到“0”个结果

当前产品无相关内容

未找到相关内容,请尝试其他搜索词