Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/eager/
Dwrap_function_test.py27 from tensorflow.python.eager import wrap_function
58 f_add = wrap_function.wrap_function(
66 f_sub = wrap_function.wrap_function(
83 f_wrapped = wrap_function.wrap_function(
103 f_wrapped = wrap_function.wrap_function(f, [x_spec, y_spec])
137 wrapped = wrap_function.wrap_function(importer, [])
176 f_wrapped = wrap_function.wrap_function(f, [])
187 f_wrapped = wrap_function.wrap_function(f, [])
212 f_wrapped = wrap_function.wrap_function(f, [])
228 g_wrapped = wrap_function.wrap_function(g, [])
[all …]
Dwrap_function_device_test.py23 from tensorflow.python.eager import wrap_function
78 wrapped_noarg_fn = wrap_function.wrap_function(
Dwrap_function.py470 def wrap_function(self, fn, signature, name=None): member in WrappedGraph
560 def wrap_function(fn, signature, name=None): function
650 wrapped_import = wrap_function(_imports_graph_def, [])
DBUILD152 ":wrap_function",
968 name = "wrap_function",
969 srcs = ["wrap_function.py"],
989 ":wrap_function",
1003 ":wrap_function",
/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()
/external/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()
/external/tensorflow/tensorflow/python/saved_model/
Dload_v1_in_v2.py25 from tensorflow.python.eager import wrap_function
147wrap_function._get_element_from_tensor_info(input_spec, wrapped.graph) # pylint: disable=protecte…
224 wrapped = wrap_function.wrap_function(
Dfunction_serialization.py160 def wrap_function(*args): function
166 None, wrap_function, args=tuple(args), kwargs={},
DBUILD448 "//tensorflow/python/eager:wrap_function",
Dload_test.py35 from tensorflow.python.eager import wrap_function
246 f_wrapped = wrap_function.wrap_function(
/external/tensorflow/tensorflow/python/training/saving/
Dfunctional_saver_test.py26 from tensorflow.python.eager import wrap_function
110 wrapped = wrap_function.wrap_function(
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert.py35 from tensorflow.python.eager import wrap_function
1006 rebuilt_func = wrap_function.function_from_graph_def(
1059 self._converted_func = wrap_function.function_from_graph_def(
1220 reset_converted_func = wrap_function.function_from_graph_def(
/external/tensorflow/tensorflow/python/ops/
Dstandard_ops.py58 from tensorflow.python.eager import wrap_function
/external/tensorflow/tensorflow/python/keras/layers/
Dnormalization_test.py26 from tensorflow.python.eager import wrap_function
375 wrapped_fn = wrap_function.wrap_function(my_func, [])
/external/tensorflow/tensorflow/python/framework/
Dconvert_to_constants.py45 wrap_function = lazy_loader.LazyLoader( variable
1008 new_func = wrap_function.function_from_graph_def(output_graph_def,
Dops_test.py36 from tensorflow.python.eager import wrap_function
3420 wrap_function.function_from_graph_def(graph_def, [], ["output"])
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_scope_test.py28 from tensorflow.python.eager import wrap_function
63 wrapped = wrap_function.wrap_function(graph_function, [self])
Dcontrol_flow_ops_py_test.py42 from tensorflow.python.eager import wrap_function
1447 pruned_cond = wrap_function.wrap_function(pruned_cond, [])
1455 pruned_nested_cond = wrap_function.wrap_function(pruned_nested_cond, [])
1521 pruned_while = wrap_function.wrap_function(pruned_while, [])
1529 pruned_nested_while = wrap_function.wrap_function(pruned_nested_while, [])
Dlookup_ops_test.py35 from tensorflow.python.eager import wrap_function
364 wrapped = wrap_function.wrap_function(f, [])
DBUILD1492 "//tensorflow/python/eager:wrap_function",
/external/tensorflow/tensorflow/python/data/ops/
Ddataset_ops.py83 wrap_function = lazy_loader.LazyLoader( variable
297 variant_function = wrap_function.function_from_graph_def(
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.pbtxt2532 name: "wrap_function"
/external/tensorflow/tensorflow/python/
DBUILD4390 "//tensorflow/python/eager:wrap_function",