Searched refs:numpy_function (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | script_ops_test.py | 23 from tensorflow.python.ops.script_ops import numpy_function
35 actual_result = script_ops.numpy_function(plus, [1, 2], dtypes.int32)
49 return numpy_function(plus, [a, b], dtypes.int32, stateful=False)
74 return numpy_function(plus, [a, b], dtypes.int32, stateful=True)
|
| H A D | script_ops.py | 689 def numpy_function(func, inp, Tout, stateful=True, name=None): function
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/service/ |
| H A D | metadata_test.py | 197 return script_ops.numpy_function(
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/ |
| H A D | dataset_ops.py | 1009 return script_ops.numpy_function(generator_state.get_next_id, args,
1078 flat_values = script_ops.numpy_function(generator_py_func,
1136 return script_ops.numpy_function(finalize_py_func, [iterator_id_t],
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/framework/ |
| H A D | op_callbacks_test.py | 126 instrumented_output = script_ops.numpy_function(
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v2/ |
| H A D | tensorflow.pbtxt | 856 name: "numpy_function"
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/api/golden/v1/ |
| H A D | tensorflow.pbtxt | 1756 name: "numpy_function"
|
| /aosp_15_r20/external/tensorflow/ |
| H A D | RELEASE.md | 5795 `tf.py_func` and `tf.numpy_function`.
|