Searched refs:PythonUDF (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/distributed/rpc/ |
| H A D | api.py | 40 PythonUDF,
684 PythonUDF(func, args, kwargs)
740 PythonUDF(func, args, kwargs)
|
| H A D | internal.py | 284 PythonUDF = collections.namedtuple("PythonUDF", ["func", "args", "kwargs"]) variable
|
| /aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/ |
| H A D | rpc_test.py | 25 PythonUDF,
152 PythonUDF(my_tensor_function, (torch.ones(2, 2), torch.ones(2, 2)), None)
|