Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dpy_func_test.py74 f = script_ops.eager_py_func(raise_exception, [], [])
77 f = script_ops.eager_py_func(raise_exception, [], [])
132 script_ops.eager_py_func(np_func, [x, y], [dtypes.float32]))
502 _ = script_ops.eager_py_func(lambda x: x + 1, [c], [dtypes.float32])
509 script_ops.eager_py_func(
525 _ = script_ops.eager_py_func(lambda x: x + 1, [c], [dtypes.float32])
532 script_ops.eager_py_func(
546 output = script_ops.eager_py_func(matmul, inp=[a, x], Tout=dtypes.int32)
561 output = script_ops.eager_py_func(matmul, inp=[a, x], Tout=dtypes.float32)
570 output = script_ops.eager_py_func(matmul, inp=[a, x], Tout=dtypes.float32)
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Derror_interpolation_test.py308 y = script_ops.eager_py_func(lambda: [[1, 2, 3]], (), dtypes.int32)
323 y = script_ops.eager_py_func(lambda: [[1, 2, 3]], (), dtypes.int32)
361 y = script_ops.eager_py_func(lambda: [[1, 2, 3]], (), dtypes.int32)
H A Dauto_control_deps_test.py79 script_ops.eager_py_func(side_effect_one, [1], [dtypes.int32])
80 script_ops.eager_py_func(side_effect_two, [1], [dtypes.int32])
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dscript_ops.py373 result = gen_script_ops.eager_py_func(
425 def eager_py_func(func, inp, Tout, name=None): function
H A Dscript_ops_test.py102 actual_result = script_ops.eager_py_func(plus, [v1, 2], dtypes.int32)
H A Dcontrol_flow_ops_test.py1749 script_ops.eager_py_func(side_effect_py_fn, [], [dtypes.int32])
1822 return script_ops.eager_py_func(wait_then_tick_py_fn, [i],
/aosp_15_r20/external/tensorflow/tensorflow/python/autograph/utils/
H A Dpy_func.py128 return script_ops.eager_py_func(f_wrapper, tensor_args, return_dtypes)
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Dstructured_function.py226 return script_ops.eager_py_func(
H A Ddataset_ops.py1121 return script_ops.eager_py_func(
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dremote_execution_test.py145 py_func_op = script_ops.eager_py_func(
H A Dcore_test.py580 script_ops.eager_py_func(simple_fn, [v.handle], dtypes.float32)
595 return script_ops.eager_py_func(simple_fn, [v], dtypes.float32)
H A Dfunction_test.py4871 return script_ops.eager_py_func(
/aosp_15_r20/external/tensorflow/tensorflow/python/data/kernel_tests/
H A Dcheckpoint_test.py566 lambda x: script_ops.eager_py_func(fn, [x], dtypes.int64))
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Ddata_adapter.py486 flat_out = script_ops.eager_py_func(py_method, [indices], flat_dtypes)