Home
last modified time | relevance | path

Searched refs:eager_py_func (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/python/kernel_tests/
Dpy_func_test.py76 f = script_ops.eager_py_func(raise_exception, [], [])
79 f = script_ops.eager_py_func(raise_exception, [], [])
134 script_ops.eager_py_func(np_func, [x, y], [dtypes.float32]))
504 _ = script_ops.eager_py_func(lambda x: x + 1, [c], [dtypes.float32])
511 script_ops.eager_py_func(
527 _ = script_ops.eager_py_func(lambda x: x + 1, [c], [dtypes.float32])
534 script_ops.eager_py_func(
548 output = script_ops.eager_py_func(matmul, inp=[a, x], Tout=dtypes.int32)
563 output = script_ops.eager_py_func(matmul, inp=[a, x], Tout=dtypes.float32)
572 output = script_ops.eager_py_func(matmul, inp=[a, x], Tout=dtypes.float32)
[all …]
/external/tensorflow/tensorflow/python/autograph/utils/
Dpy_func.py132 return script_ops.eager_py_func(f_wrapper, tensor_args, return_dtypes)
/external/tensorflow/tensorflow/python/ops/
Dscript_ops.py343 result = gen_script_ops.eager_py_func(
433 def eager_py_func(func, inp, Tout, name=None): function
/external/tensorflow/tensorflow/python/eager/
Dremote_execution_test.py149 py_func_op = script_ops.eager_py_func(
Dcore_test.py579 script_ops.eager_py_func(simple_fn, [v.handle], dtypes.float32)
593 return script_ops.eager_py_func(simple_fn, [v], dtypes.float32)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcheckpoint_test.py546 lambda x: script_ops.eager_py_func(fn, [x], dtypes.int64))
/external/tensorflow/tensorflow/python/keras/engine/
Ddata_adapter.py507 flat_out = script_ops.eager_py_func(py_method, [indices], flat_dtypes)