tvm_book.relay#

Classes#

FuseL2Normalize

融合 torch.nn.functional.normalize

L2NormalizeONNX

L2NormalizeTorch

Module Contents#

class tvm_book.relay.FuseL2Normalize[源代码]#

融合 torch.nn.functional.normalize

transform_module(mod, ctx)[源代码]#
class tvm_book.relay.L2NormalizeONNX[源代码]#

Bases: tvm.relay.dataflow_pattern.DFPatternCallback

callback(pre, post, node_map)[源代码]#
broadcast_to[源代码]#
divide[源代码]#
multiply[源代码]#
pattern[源代码]#
sqrt[源代码]#
sum[源代码]#
x[源代码]#
class tvm_book.relay.L2NormalizeTorch[源代码]#

Bases: tvm.relay.dataflow_pattern.DFPatternCallback

callback(pre, post, node_map)[源代码]#
abs[源代码]#
broadcast_to_like[源代码]#
clip[源代码]#
divide[源代码]#
n1[源代码]#
n2[源代码]#
pattern[源代码]#
power[源代码]#
sqrt[源代码]#
sum[源代码]#
x[源代码]#