MCPcopy Create free account
hub / github.com/apache/tvm / main

Method main

tests/python/relax/test_transform_fuse_tir.py:697–701  ·  view source on GitHub ↗
(x: R.Tensor((2, 3), "float32"))

Source from the content-addressed store, hash-verified

695 class Module:
696 @R.function
697 def main(x: R.Tensor((2, 3), "float32")):
698 with R.dataflow():
699 y = R.call_dps_packed("func_packed_dps", x, R.Tensor((2, 3), "float32"))
700 R.output(y)
701 return y
702
703 # FuseTIR should do no change to it.
704 _check(Module, Module)

Callers

nothing calls this directly

Calls 11

TensorMethod · 0.80
dataflowMethod · 0.80
call_dps_packedMethod · 0.80
outputMethod · 0.80
ShapeMethod · 0.80
fused_reshapeMethod · 0.45
fused_funcMethod · 0.45
fused_add_exp_squeezeMethod · 0.45
fused_sumsMethod · 0.45

Tested by

no test coverage detected