Searched refs:my_tensor_function (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
| H A D | rpc_test.py | 152 PythonUDF(my_tensor_function, (torch.ones(2, 2), torch.ones(2, 2)), None)
251 def my_tensor_function(a, b): function
778 my_tensor_function,
787 my_tensor_function,
796 my_tensor_function,
2569 my_tensor_function,
2572 self.assertEqual(ret, my_tensor_function(torch.ones(n, n), torch.ones(n, n)))
2582 my_tensor_function,
2590 val, my_tensor_function(torch.ones(j, j), torch.ones(j, j))
2618 m.set(my_tensor_function(torch.ones(2, 2), torch.ones(2, 2)))
|