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

Method main

tests/python/relax/test_frontend_from_exported_program.py:1657–1665  ·  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

1655 class expected_min1:
1656 @R.function
1657 def main(
1658 inp_0: R.Tensor((10, 10), dtype="float32"),
1659 inp_1: R.Tensor((10, 10), dtype="float32"),
1660 ) -> R.Tuple(R.Tensor((10, 10), dtype="float32")):
1661 with R.dataflow():
1662 lv: R.Tensor((10, 10), dtype="float32") = R.minimum(inp_0, inp_1)
1663 gv: R.Tuple(R.Tensor((10, 10), dtype="float32")) = (lv,)
1664 R.output(gv)
1665 return gv
1666
1667 verify_model(Min1(), example_args1, {}, expected_min1)
1668

Callers

nothing calls this directly

Calls 4

TensorMethod · 0.80
dataflowMethod · 0.80
outputMethod · 0.80
minimumMethod · 0.45

Tested by

no test coverage detected