Searched refs:wrap_py_func (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/python/autograph/utils/ |
| D | py_func_test.py | 35 result = py_func.wrap_py_func(test_fn, dtypes.int32,
38 result = py_func.wrap_py_func(test_fn, dtypes.int32, (1, 1, 1))
40 result = py_func.wrap_py_func(
56 result = py_func.wrap_py_func(test_fn, dtypes.int32, (7, TestClass()))
58 result = py_func.wrap_py_func(test_fn, dtypes.int32,
73 result = py_func.wrap_py_func(test_fn, dtypes.int32, (7, TestClass(5)), {
78 result = py_func.wrap_py_func(test_fn, dtypes.int32,
93 result = py_func.wrap_py_func(test_fn, None, (5,), use_dummy_return=True)
96 result = py_func.wrap_py_func(
|
| D | __init__.py | 23 from tensorflow.python.autograph.utils.py_func import wrap_py_func
|
| D | py_func.py | 38 def wrap_py_func(f, return_dtypes, args, kwargs=None, use_dummy_return=False): function
|
| /external/tensorflow/tensorflow/python/autograph/operators/ |
| D | py_builtins.py | 358 return py_func.wrap_py_func(
|