Searched refs:my_class_method (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/pytorch/test/dynamo/ |
| H A D | test_decorators.py | 505 def my_class_method(cls, arg1): member in DecoratorTests.test_class_methods.A
516 def my_class_method(self, arg1): member in DecoratorTests.test_class_methods.B
517 return super().my_class_method(arg1)
524 def my_class_method(cls, arg1): member in DecoratorTests.test_class_methods.C
525 return super().my_class_method(arg1)
533 v1 = a.my_class_method(1)
534 v2 = A.my_class_method(2)
538 v6 = b.my_class_method(6)
540 v8 = c.my_class_method(8)
541 v9 = C.my_class_method(9)
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
| H A D | rpc_test.py | 189 def my_class_method(cls, d, e): member in MyClass
1329 expected.my_class_method(2, torch.zeros(2, 2)),
1330 rref.rpc_sync().my_class_method(2, torch.zeros(2, 2))
1333 expected.my_class_method(2, torch.ones(3, 3)),
1334 rref.rpc_async().my_class_method(2, torch.ones(3, 3)).wait()
1337 expected.my_class_method(2, torch.ones(4, 4)),
1338 rref.remote().my_class_method(2, torch.ones(4, 4)).to_here()
2533 worker_name(dst_rank), MyClass.my_class_method, args=(n, n + 1)
2535 self.assertEqual(ret, MyClass.my_class_method(n, n + 1))
|