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

Method foo

tests/python/relax/test_transform_cse.py:49–55  ·  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

47 class Expected:
48 @R.function
49 def foo(x: R.Tensor((2, 3), dtype="float32"), y: R.Tensor((2, 3), dtype="float32")):
50 with R.dataflow():
51 lv0 = R.add(x, y)
52 lv1 = lv0
53 gv = R.multiply(lv0, lv0)
54 R.output(gv)
55 return gv
56
57 verify(Before, Expected)
58

Callers

nothing calls this directly

Calls 10

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

Tested by

no test coverage detected