MCPcopy Index your code
hub / github.com/apache/tvm / foo

Method foo

tests/python/relax/test_transform_cse.py:38–44  ·  view source on GitHub ↗
(x: R.Tensor((2, 3), dtype="float32"), y: R.Tensor((2, 3), dtype="float32"))

Source from the content-addressed store, hash-verified

36 class Before:
37 @R.function
38 def foo(x: R.Tensor((2, 3), dtype="float32"), y: R.Tensor((2, 3), dtype="float32")):
39 with R.dataflow():
40 lv0 = R.add(x, y)
41 lv1 = R.add(x, y)
42 gv = R.multiply(lv0, lv1)
43 R.output(gv)
44 return gv
45
46 @I.ir_module(s_tir=True)
47 class Expected:

Callers

nothing calls this directly

Calls 14

TensorMethod · 0.80
dataflowMethod · 0.80
outputMethod · 0.80
call_packedMethod · 0.80
match_castMethod · 0.80
barFunction · 0.70
addMethod · 0.45
multiplyMethod · 0.45
zerosMethod · 0.45
arangeMethod · 0.45
printMethod · 0.45
reshapeMethod · 0.45

Tested by

no test coverage detected