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

Method main

tests/python/relax/test_frontend_from_exported_program.py:1637–1645  ·  view source on GitHub ↗
(
            inp_0: R.Tensor((10, 10), dtype="float32"),
            inp_1: R.Tensor((10, 10), dtype="float32"),
        )

Source from the content-addressed store, hash-verified

1635 class expected_max1:
1636 @R.function
1637 def main(
1638 inp_0: R.Tensor((10, 10), dtype="float32"),
1639 inp_1: R.Tensor((10, 10), dtype="float32"),
1640 ) -> R.Tuple(R.Tensor((10, 10), dtype="float32")):
1641 with R.dataflow():
1642 lv: R.Tensor((10, 10), dtype="float32") = R.maximum(inp_0, inp_1)
1643 gv: R.Tuple(R.Tensor((10, 10), dtype="float32")) = (lv,)
1644 R.output(gv)
1645 return gv
1646
1647 verify_model(Max1(), example_args1, {}, expected_max1)
1648

Callers

nothing calls this directly

Calls 4

TensorMethod · 0.80
dataflowMethod · 0.80
outputMethod · 0.80
maximumMethod · 0.45

Tested by

no test coverage detected