Home
last modified time | relevance | path

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

12

/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dwrap_function_test.py23 from tensorflow.python.eager import wrap_function
54 f_add = wrap_function.wrap_function(
62 f_sub = wrap_function.wrap_function(
79 f_wrapped = wrap_function.wrap_function(
99 f_wrapped = wrap_function.wrap_function(f, [x_spec, y_spec])
133 wrapped = wrap_function.wrap_function(importer, [])
172 f_wrapped = wrap_function.wrap_function(f, [])
183 f_wrapped = wrap_function.wrap_function(f, [])
208 f_wrapped = wrap_function.wrap_function(f, [])
224 g_wrapped = wrap_function.wrap_function(g, [])
[all …]
H A Dwrap_function_device_test.py19 from tensorflow.python.eager import wrap_function
74 wrapped_noarg_fn = wrap_function.wrap_function(
H A Dwrap_function.py468 def wrap_function(self, fn, signature, name=None): member in WrappedGraph
558 def wrap_function(fn, signature, name=None): function
655 wrapped_import = wrap_function(_imports_graph_def, [])
H A DBUILD152 ":wrap_function",
1069 name = "wrap_function",
1070 srcs = ["wrap_function.py"],
1090 ":wrap_function",
1104 ":wrap_function",
/aosp_15_r20/external/ComputeLibrary/src/graph/backends/CL/
H A DCLFunctionsFactory.cpp163 auto wrap_function = std::make_unique<CPPWrapperFunction>(); in create_detection_output_layer() local
165 wrap_function->register_function(std::move(func)); in create_detection_output_layer()
166 wrap_function->register_tensor(input0); in create_detection_output_layer()
167 wrap_function->register_tensor(input1); in create_detection_output_layer()
168 wrap_function->register_tensor(input2); in create_detection_output_layer()
169 wrap_function->register_tensor(output); in create_detection_output_layer()
171 return std::move(wrap_function); in create_detection_output_layer()
216 auto wrap_function = std::make_unique<CPPWrapperFunction>(); in create_detection_post_process_layer() local
218 wrap_function->register_function(std::move(func)); in create_detection_post_process_layer()
219 wrap_function->register_tensor(input0); in create_detection_post_process_layer()
[all …]
/aosp_15_r20/external/tensorflow/tensorflow/python/saved_model/
H A Dload_v1_in_v2.py21 from tensorflow.python.eager import wrap_function
150 …wrap_function._get_element_from_tensor_info(input_spec, wrapped.graph) # pylint: disable=protecte…
227 wrapped = wrap_function.wrap_function(
H A Dfunction_serialization.py147 def wrap_function(*args): function
153 None, wrap_function, args=tuple(args), kwargs={},
H A DBUILD475 "//tensorflow/python/eager:wrap_function",
H A Dload_test.py39 from tensorflow.python.eager import wrap_function
256 f_wrapped = wrap_function.wrap_function(
/aosp_15_r20/external/rust/crates/quiche/deps/boringssl/src/util/fipstools/cavp/
Dcavp_keywrap_test.cc110 auto wrap_function = AESKeyWrap; in TestCipher() local
112 wrap_function = AESKeyWrapWithPadding; in TestCipher()
116 if (!wrap_function(&result, ctx->encrypt, key, in)) { in TestCipher()
/aosp_15_r20/external/rust/android-crates-io/crates/quiche/deps/boringssl/src/util/fipstools/cavp/
Dcavp_keywrap_test.cc110 auto wrap_function = AESKeyWrap; in TestCipher() local
112 wrap_function = AESKeyWrapWithPadding; in TestCipher()
116 if (!wrap_function(&result, ctx->encrypt, key, in)) { in TestCipher()
/aosp_15_r20/external/tensorflow/tensorflow/python/trackable/
H A Dresource_test.py17 from tensorflow.python.eager import wrap_function
39 wrapped = wrap_function.wrap_function(graph_function, [self])
/aosp_15_r20/external/tensorflow/tensorflow/python/checkpoint/
H A Dfunctional_saver_test.py24 from tensorflow.python.eager import wrap_function
106 wrapped = wrap_function.wrap_function(
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/python/
H A Dquantize_model.py36 from tensorflow.python.eager import wrap_function
405 func: wrap_function.WrappedFunction,
H A DBUILD149 "//tensorflow/python/eager:wrap_function",
/aosp_15_r20/external/tensorflow/tensorflow/python/compiler/tensorrt/
H A Dtrt_convert.py34 from tensorflow.python.eager import wrap_function
914 new_func = wrap_function.function_from_graph_def(
1574 reset_converted_func = wrap_function.function_from_graph_def(
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Dstandard_ops.py53 from tensorflow.python.eager import wrap_function
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/layers/normalization/
H A DBUILD111 "//tensorflow/python/eager:wrap_function",
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/variables/
H A DBUILD121 "//tensorflow/python/eager:wrap_function",
H A Dvariable_scope_test.py24 from tensorflow.python.eager import wrap_function
59 wrapped = wrap_function.wrap_function(graph_function, [self])
/aosp_15_r20/external/tensorflow/tensorflow/python/framework/
H A Dconvert_to_constants.py44 wrap_function = lazy_loader.LazyLoader( variable
1087 new_func = wrap_function.function_from_graph_def(output_graph_def,
H A Dops_test.py36 from tensorflow.python.eager import wrap_function
3506 wrap_function.function_from_graph_def(graph_def, [], ["output"])
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/control_flow/
H A Dcontrol_flow_ops_py_test.py38 from tensorflow.python.eager import wrap_function
1462 pruned_cond = wrap_function.wrap_function(pruned_cond, [])
1470 pruned_nested_cond = wrap_function.wrap_function(pruned_nested_cond, [])
1536 pruned_while = wrap_function.wrap_function(pruned_while, [])
1544 pruned_nested_while = wrap_function.wrap_function(pruned_nested_while, [])
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/data_structures/
H A Dlookup_ops_test.py34 from tensorflow.python.eager import wrap_function
429 wrapped = wrap_function.wrap_function(f, [])
3380 wrapped = wrap_function.wrap_function(f, [])
/aosp_15_r20/external/tensorflow/tensorflow/python/data/ops/
H A Ddataset_ops.py81 wrap_function = lazy_loader.LazyLoader( variable
403 variant_function = wrap_function.function_from_graph_def(

12