xref: /aosp_15_r20/external/pytorch/test/jit/myfunction_b.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45)
1r"""
2Helper function used in test_decorator.py. We define it in a
3separate file on purpose to test that the names in different modules
4are resolved correctly.
5"""
6
7from jit.mydecorator import my_decorator
8
9
10@my_decorator
11def my_function_b(x: float) -> float:
12    return my_function_c(x) + 2
13
14
15def my_function_c(x: float) -> float:
16    return x + 3
17