(
inp_0: R.Tensor((10, 10), dtype="float32"),
inp_1: R.Tensor((10, 10), dtype="float32"),
)
| 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 |