tvm_book.transforms.common#

Module Contents#

Classes#

FuseTransform

替换融合函数为全局函数

OutputTransform

将中间全局函数节点转换为输出节点

class tvm_book.transforms.common.FuseTransform[源代码]#

替换融合函数为全局函数

reset()[源代码]#
transform_function(func, mod, ctx)[源代码]#
class tvm_book.transforms.common.OutputTransform[源代码]#

将中间全局函数节点转换为输出节点

reset()[源代码]#
transform_function(func, mod, ctx)[源代码]#