Home
last modified time | relevance | path

Searched refs:python_udf (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/torch/distributed/rpc/
H A Dinternal.py198 def _run_function(python_udf): argument
207 if isinstance(python_udf, AttributeError):
208 raise python_udf
209 result = python_udf.func(*python_udf.args, **python_udf.kwargs)
/aosp_15_r20/external/pytorch/torch/testing/_internal/distributed/rpc/
H A Drpc_test.py157 python_udf = _internal_rpc_pickler.deserialize(obj[0], obj[1])
158 result = python_udf.func(python_udf.args[0], python_udf.args[1])