Searched refs:wrap_py_func (Results 1 – 4 of 4) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/autograph/utils/ |
| H A D | py_func_test.py | 31 result = py_func.wrap_py_func(test_fn, dtypes.int32,
34 result = py_func.wrap_py_func(test_fn, dtypes.int32, (1, 1, 1))
36 result = py_func.wrap_py_func(
52 result = py_func.wrap_py_func(test_fn, dtypes.int32, (7, TestClass()))
54 result = py_func.wrap_py_func(test_fn, dtypes.int32,
69 result = py_func.wrap_py_func(test_fn, dtypes.int32, (7, TestClass(5)), {
74 result = py_func.wrap_py_func(test_fn, dtypes.int32,
89 result = py_func.wrap_py_func(test_fn, None, (5,), use_dummy_return=True)
92 result = py_func.wrap_py_func(
|
| H A D | __init__.py | 19 from tensorflow.python.autograph.utils.py_func import wrap_py_func
|
| H A D | py_func.py | 34 def wrap_py_func(f, return_dtypes, args, kwargs=None, use_dummy_return=False): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/autograph/operators/ |
| H A D | py_builtins.py | 409 return py_func.wrap_py_func(
|