Home
last modified time | relevance | path

Searched refs:concrete_function (Results 1 – 25 of 34) sorted by relevance

12

/external/tensorflow/tensorflow/python/saved_model/
Dfunction_serialization.py55 def serialize_concrete_function(concrete_function, node_ids): argument
59 for capture in concrete_function.captured_inputs:
71 concrete_function.structured_outputs)
74 concrete_function.structured_input_signature))
81 def serialize_bare_concrete_function(concrete_function): argument
85 concrete_function_name=concrete_function.name,
86 allowed_positional_arguments=concrete_function._num_positional_args,
87 argument_keywords=concrete_function._arg_keywords)
88 if concrete_function._pre_initialized_function_spec is not None:
91 concrete_function._pre_initialized_function_spec))
[all …]
Dsignature_serialization.py49 def _valid_signature(concrete_function): argument
51 if not concrete_function.outputs:
57 _validate_inputs(concrete_function)
58 _normalize_outputs(concrete_function.structured_outputs, "unused", "unused")
64 def _validate_inputs(concrete_function): argument
68 concrete_function.structured_input_signature)):
75 def _get_signature_name_changes(concrete_function): argument
80 concrete_function.function_def.signature.input_arg,
81 concrete_function.graph.inputs):
240 def _add_signature(self, name, concrete_function): argument
[all …]
Dfunction_deserialization.py170 concrete_function = concrete_functions[
173 concrete_function._arg_keywords = (
175 concrete_function._num_positional_args = (
180 concrete_function._set_function_spec(function_spec)
182 concrete_function.add_to_graph()
183 return concrete_function
290 concrete_function = concrete_functions[function_name]
291 positional, keyword = concrete_function.structured_input_signature
Dsave.py195 def _maybe_uncache_variable_captures(self, concrete_function): argument
196 if concrete_function in self._wrapped_functions:
197 return self._wrapped_functions[concrete_function]
198 for capture in concrete_function.captured_inputs:
200 if concrete_function not in self._wrapped_functions:
201 wrapped = self._wrapped_functions[concrete_function] = (
202 function_serialization.wrap_cached_variables(concrete_function))
204 return concrete_function
965 for concrete_function in saveable_view.concrete_and_gradient_functions:
966 name = compat.as_text(concrete_function.name)
[all …]
Dload.py112 def __init__(self, concrete_function): argument
114 self.__dict__.update(vars(concrete_function))
364 concrete_function = self._concrete_functions[concrete_function_name]
367 function_saved_model_utils.restore_captures(concrete_function, inputs)
713 outer_graph = self._concrete_functions[proto.concrete_function].graph
/external/tensorflow/tensorflow/cc/saved_model/experimental/public/
DBUILD15 name = "concrete_function",
17 "concrete_function.h",
22 "//tensorflow/c/experimental/saved_model/public:concrete_function",
33 ":concrete_function",
54 ":concrete_function",
/external/tensorflow/tensorflow/python/compiler/mlir/
Dmlir_test.py125 concrete_function = sqr.get_concrete_function(
127 mlir_module = mlir.convert_function(concrete_function, show_debug_info=True)
142 concrete_function = caller.get_concrete_function(
144 mlir_module = mlir.convert_function(concrete_function)
155 concrete_function = logging.get_concrete_function()
156 mlir_module = mlir.convert_function(concrete_function, pass_pipeline='')
Dmlir.py50 def convert_function(concrete_function, argument
90 return pywrap_mlir.import_function(concrete_function, pass_pipeline,
/external/tensorflow/tensorflow/python/eager/
Dfunction_saved_model_utils.py217 def restore_captures(concrete_function, inputs): argument
241 concrete_function.set_variables(bound_variables)
244 bound_inputs, concrete_function.inputs[-len(bound_inputs):]):
250 concrete_function, internal_capture))
253 concrete_function.graph.replace_capture(bound_input, internal_capture)
269 concrete_function.graph.capture(bound_input)
276 concrete_function.set_external_captures(captured_inputs_list)
Dcancellation.py37 def get_cancelable_function(self, concrete_function): argument
39 return concrete_function._experimental_with_cancellation_manager(self)
Ddef_function.py1148 for concrete_function in concrete_functions:
1149 signature = concrete_function.structured_input_signature
/external/tensorflow/tensorflow/c/experimental/saved_model/internal/
DBUILD32 name = "concrete_function",
34 "concrete_function.cc",
37 "//tensorflow/c/experimental/saved_model/public:concrete_function.h",
54 "//tensorflow/c/experimental/saved_model/core:concrete_function",
74 ":concrete_function",
78 "//tensorflow/c/experimental/saved_model/core:concrete_function",
88 "//tensorflow/c/experimental/saved_model/core:concrete_function",
99 "//tensorflow/c/experimental/saved_model/core:concrete_function",
148 ":concrete_function",
376 "//tensorflow/c/experimental/saved_model/public:concrete_function",
/external/tensorflow/tensorflow/c/experimental/saved_model/public/
DBUILD25 "concrete_function.h",
45 ":concrete_function",
58 name = "concrete_function",
59 actual = "//tensorflow/c/experimental/saved_model/internal:concrete_function",
/external/tflite-support/tensorflow_lite_support/custom_ops/python/
Dragged_tensor_to_tensor_test.py35 concrete_function = ragged_tensor_function.get_concrete_function()
38 [concrete_function])
/external/tensorflow/tensorflow/python/
Dpywrap_mlir.py42 def import_function(concrete_function, pass_pipeline, show_debug_info): argument
46 str(concrete_function.function_def).encode('utf-8'),
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dtf_saved_model_api.cc297 std::unique_ptr<TFConcreteFunction> concrete_function; in Load() local
302 /*out=*/&concrete_function)); in Load()
303 revived_objects.concrete_functions.Insert(std::move(concrete_function)); in Load()
DBUILD24 name = "concrete_function",
26 "concrete_function.h",
49 ":concrete_function",
158 ":concrete_function",
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.mlir.experimental.pbtxt5 …argspec: "args=[\'concrete_function\', \'pass_pipeline\', \'show_debug_info\'], varargs=None, keyw…
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.mlir.experimental.pbtxt5 …argspec: "args=[\'concrete_function\', \'pass_pipeline\', \'show_debug_info\'], varargs=None, keyw…
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_cli.py209 for index, concrete_function in enumerate(concrete_functions, 1):
211 if concrete_function.structured_input_signature:
212 args, kwargs = concrete_function.structured_input_signature
213 elif concrete_function._arg_keywords: # pylint: disable=protected-access
216 args = concrete_function._arg_keywords # pylint: disable=protected-access
/external/tensorflow/tensorflow/cc/saved_model/
Dfingerprinting.cc181 for (const auto& [name, concrete_function] : in RegularizeAndHashSavedObjectGraph()
/external/tensorflow/tensorflow/c/experimental/saved_model/
DREADME.md53 The files fulfilling 1. are named `*.cc` (eg: `concrete_function.cc`), while
/external/tensorflow/tensorflow/python/distribute/
Dps_values.py658 self, concrete_function, internal_capture): argument
660 concrete_function.graph.replace_capture_with_deferred_capture(
666 return concrete_function.graph.deferred_external_captures[-1]
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
DBUILD202 "//tensorflow/c/experimental/saved_model/core:concrete_function",
/external/tensorflow/tensorflow/lite/g3doc/examples/convert/
Dconvert_models.md104 ### Convert concrete functions <a name="concrete_function"></a>

12