Home
last modified time | relevance | path

Searched refs:py_func (Results 1 – 25 of 27) sorted by relevance

12

/external/tensorflow/tensorflow/python/autograph/utils/
Dpy_func_test.py21 from tensorflow.python.autograph.utils import py_func
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__.py23 from tensorflow.python.autograph.utils.py_func import wrap_py_func
DBUILD26 "py_func.py",
/external/tensorflow/tensorflow/python/kernel_tests/
Dpy_func_test.py81 f = script_ops.py_func(raise_exception, [], [])
99 z = self.evaluate(script_ops.py_func(sum_func, [x, y], dtype))
109 z = self.evaluate(script_ops.py_func(sub_func, [x, y], dtype))
119 z = self.evaluate(script_ops.py_func(and_func, [x, y], dtype))
126 z = self.evaluate(script_ops.py_func(np_func, [x, y], dtypes.float32))
142 z = self.evaluate(script_ops.py_func(np_func, [x, y], [dtypes.float64]))
150 z = self.evaluate(script_ops.py_func(np_func, [x, y], dtypes.complex64))
160 y = self.evaluate(script_ops.py_func(rfft, [x], dtypes.complex64))
170 y = self.evaluate(script_ops.py_func(literal, [x], dtypes.float64))
181 script_ops.py_func(list_func, [x], [dtypes.float64] * 2))
[all …]
/external/tensorflow/tensorflow/python/framework/
Dsubscribe_test.py64 lambda t: script_ops.py_func(sub, [t], [t.dtype]))
92 (a, b), lambda t: script_ops.py_func(sub, [t], [t.dtype]))
98 [a, b], lambda t: script_ops.py_func(sub, [t], [t.dtype]))
106 }, lambda t: script_ops.py_func(sub, [t], [t.dtype]))
116 pair, lambda t: script_ops.py_func(sub, [t], [t.dtype]))
123 lambda t: script_ops.py_func(sub, [t], [t.dtype]))
140 lambda t: script_ops.py_func(sub, [t], [t.dtype]))
148 lambda t: script_ops.py_func(sub, [t], [t.dtype]))
188 c, lambda t: script_ops.py_func(sub_graph1, [t], [t.dtype]))
194 c_sub, lambda t: script_ops.py_func(sub_graph2, [t], [t.dtype]))
[all …]
Dtest_util.py1248 return script_ops.py_func(inner_f, tensor_args, [])
/external/tensorflow/tensorflow/python/ops/
Dscript_ops.py351 result = gen_script_ops.py_func(
635 def py_func(func, inp, Tout, stateful=True, name=None): function
639 py_func.__doc__ = "%s" % py_func_common.__doc__
Dstandard_ops.py83 from tensorflow.python.ops.script_ops import py_func
Dbatch_ops_test.py536 computation_delayed = script_ops.py_func(delayed_plus1,
/external/tensorflow/tensorflow/python/lib/core/
DBUILD212 srcs = ["py_func.cc"],
213 hdrs = ["py_func.h"],
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Doverride_threadpool_test.py53 lambda x: script_ops.py_func(get_thread_id, [x], dtypes.int64),
Dparallel_interleave_test.py73 return script_ops.py_func(map_py_fn, [x], x.dtype)
627 return script_ops.py_func(map_py_fn, [x], x.dtype)
674 y = script_ops.py_func(map_py_fn, [x], x.dtype)
Dscan_test.py257 return state, script_ops.py_func(py_fn, [val], dtypes.int64)
Dmap_and_batch_test.py290 lambda x: script_ops.py_func(raising_py_fn, [x], dtypes.int64),
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dmap_test.py146 return script_ops.py_func(map_py_fn, [x], x.dtype)
742 return script_ops.py_func(_map_py_func, [x], x.dtype)
780 return script_ops.py_func(
926 lambda x: script_ops.py_func(raising_py_func, [x], dtypes.int64),
1062 tids.append(script_ops.py_func(_get_tid, [], dtypes.int64))
1164 lambda x: script_ops.py_func(py_fn, [x], dtypes.int64),
1355 return script_ops.py_func(sleep, [x], x.dtype)
Dfrom_generator_test.py425 return script_ops.py_func(finalize_py_func, [], [dtypes.int64],
Diterator_test.py726 handle = script_ops.py_func(_encode_raw, [h], dtypes.string)
/external/tensorflow/tensorflow/python/autograph/operators/
Dpy_builtins.py30 from tensorflow.python.autograph.utils import py_func
358 return py_func.wrap_py_func(
/external/python/cpython2/Modules/_ctypes/
D_ctypes_test.c398 PyObject *py_func(PyObject *self, PyObject *args) in py_func() function
468 {"func", py_func, METH_NOARGS},
/external/python/cpython2/Modules/_sqlite/
Dconnection.c654 PyObject* py_func; in _pysqlite_func_callback() local
664 py_func = (PyObject*)sqlite3_user_data(context); in _pysqlite_func_callback()
668 py_retval = PyObject_CallObject(py_func, args); in _pysqlite_func_callback()
/external/python/cpython3/Modules/_sqlite/
Dconnection.c592 PyObject* py_func; in _pysqlite_func_callback() local
600 py_func = (PyObject*)sqlite3_user_data(context); in _pysqlite_func_callback()
604 py_retval = PyObject_CallObject(py_func, args); in _pysqlite_func_callback()
/external/python/cpython3/Modules/_ctypes/
D_ctypes_test.c573 PyObject *py_func(PyObject *self, PyObject *args) in py_func() function
628 {"func", py_func, METH_NOARGS},
/external/tensorflow/tensorflow/tools/def_file_filter/
Dsymbols_pybind.txt56 [//tensorflow/python:py_func_lib] # py_func
/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/optimization/
Dmap_vectorization_test.py577 script_ops.py_func(sleep, [x], x.dtype, stateful=False), ())
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt1812 name: "py_func"

12