Method main

tests/python/relax/test_transform_legalize_ops_binary.py:799–801

(x: R.Tensor((1, 2, 3), "float32"), y: R.Tensor((4, 3, 2, 1), "float32"))

    class Atan2:
        @R.function
        def main(x: R.Tensor((1, 2, 3), "float32"), y: R.Tensor((4, 3, 2, 1), "float32")) -> R.Tensor((4, 3, 2, 3), "float32"):
            gv: R.Tensor((4, 3, 2, 3), "float32") = R.atan2(x, y)
            return gv

    @tvm.script.ir_module
    class Expected: