(
inp_0: R.Tensor((1, 3, 10, 10), dtype="float32"),
)
| 603 | class Expected_log1p: |
| 604 | @R.function |
| 605 | def main( |
| 606 | inp_0: R.Tensor((1, 3, 10, 10), dtype="float32"), |
| 607 | ) -> R.Tuple(R.Tensor((1, 3, 10, 10), dtype="float32")): |
| 608 | with R.dataflow(): |
| 609 | lv: R.Tensor((1, 3, 10, 10), dtype="float32") = R.log( |
| 610 | R.add(inp_0, R.const(1, "float32")) |
| 611 | ) |
| 612 | gv: R.Tuple(R.Tensor((1, 3, 10, 10), dtype="float32")) = (lv,) |
| 613 | R.output(gv) |
| 614 | return gv |
| 615 | |
| 616 | verify_model(Log1p(), example_args, {}, Expected_log1p) |
| 617 |