xref: /aosp_15_r20/external/pytorch/test/dynamo/mock_modules/mock_module2.py (revision da0073e96a02ea20f0ac840b70461e3646d07c45) 
1# from . import mock_module3
2import torch
3
4from . import mock_module3
5
6
7class Class1:
8    def __init__(self, x, y):
9        self.x = x
10        self.y = y
11
12    def method2(self, x):
13        return mock_module3.method1([], x)
14
15
16def method1(x, y):
17    torch.ones(1, 1)
18    x.append(y)
19    return x
20