Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/python/saved_model/
Dfunction_serialization.py60 def serialize_concrete_function(concrete_function, node_ids, coder): argument
64 for capture in concrete_function.captured_inputs:
74 % (concrete_function.name, capture))
77 concrete_function.structured_outputs)
79 coder.encode_structure(concrete_function.structured_input_signature))
86 def serialize_bare_concrete_function(concrete_function, name_map): argument
89 name = name_map.get(compat.as_text(concrete_function.name),
90 concrete_function.name)
93 allowed_positional_arguments=concrete_function._num_positional_args,
94 argument_keywords=concrete_function._arg_keywords)
[all …]
Dsignature_serialization.py52 def _valid_signature(concrete_function): argument
54 if not concrete_function.outputs:
60 _validate_inputs(concrete_function)
61 _normalize_outputs(concrete_function.structured_outputs, "unused", "unused")
67 def _validate_inputs(concrete_function): argument
71 concrete_function.structured_input_signature)):
76 def _get_signature_name_changes(concrete_function): argument
81 concrete_function.function_def.signature.input_arg,
82 concrete_function.graph.inputs):
237 def _add_signature(self, name, concrete_function): argument
[all …]
Dsave.py257 for concrete_function in concrete_functions:
258 if concrete_function.name not in self._seen_function_names:
259 self.concrete_functions.append(concrete_function)
260 self._seen_function_names.add(concrete_function.name)
354 for concrete_function in self.concrete_functions:
355 if not concrete_function.graph.saveable:
358 "\n".join(concrete_function.graph.saving_errors)).format(
359 name=concrete_function.name))
360 for capture in concrete_function.captured_inputs:
365 if concrete_function not in self._wrapped_functions:
[all …]
Dload.py89 def __init__(self, concrete_function): argument
91 self.__dict__.update(vars(concrete_function))
155 for name, concrete_function in self._concrete_functions.items():
158 self._concrete_functions[name] = _WrapperFunction(concrete_function)
314 concrete_function = self._concrete_functions[name]
327 …concrete_function._func_graph.structured_outputs = original_outputs # pylint: disable=protected-a…
328 … concrete_function._func_graph.structured_input_signature = ( # pylint: disable=protected-access
330 concrete_function._initialize_function_spec() # pylint: disable=protected-access
336 concrete_function = self._concrete_functions[name]
348 concrete_function._captured_inputs = bound_inputs # pylint: disable=protected-access
[all …]
Dfunction_deserialization.py166 concrete_function = concrete_functions[
169 concrete_function._arg_keywords = (
171 concrete_function._num_positional_args = (
178 concrete_function._set_function_spec(function_spec)
180 concrete_function.add_to_graph()
181 return concrete_function
280 concrete_function = concrete_functions[function_name]
281 positional, keyword = concrete_function.structured_input_signature
/external/tensorflow/tensorflow/python/compiler/mlir/
Dmlir_test.py54 concrete_function = sqr.get_concrete_function(
56 mlir_module = mlir.convert_function(concrete_function, show_debug_info=True)
71 concrete_function = caller.get_concrete_function(
73 mlir_module = mlir.convert_function(concrete_function)
84 concrete_function = logging.get_concrete_function()
85 mlir_module = mlir.convert_function(concrete_function, pass_pipeline='')
Dmlir.py54 def convert_function(concrete_function, argument
94 return pywrap_mlir.import_function(concrete_function, pass_pipeline,
/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/c/experimental/saved_model/internal/
DBUILD28 name = "concrete_function",
30 "concrete_function.cc",
33 "//tensorflow/c/experimental/saved_model/public:concrete_function.h",
50 "//tensorflow/c/experimental/saved_model/core:concrete_function",
70 ":concrete_function",
74 "//tensorflow/c/experimental/saved_model/core:concrete_function",
84 "//tensorflow/c/experimental/saved_model/core:concrete_function",
95 "//tensorflow/c/experimental/saved_model/core:concrete_function",
142 ":concrete_function",
363 "//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/tensorflow/tensorflow/python/eager/
Dcancellation.py41 def get_cancelable_function(self, concrete_function): argument
43 return concrete_function._experimental_with_cancellation_manager(self)
Ddef_function.py1165 for concrete_function in concrete_functions:
1166 signature = concrete_function.structured_input_signature
Ddef_function_test.py420 for concrete_function in concrete_functions:
421 args, kwargs = concrete_function.structured_input_signature
/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.py33 def import_function(concrete_function, pass_pipeline, show_debug_info): argument
37 str(concrete_function.function_def).encode('utf-8'),
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dtf_saved_model_api.cc274 std::unique_ptr<TFConcreteFunction> concrete_function; in Load() local
279 /*out=*/&concrete_function)); in Load()
280 revived_objects.concrete_functions.Insert(std::move(concrete_function)); in Load()
DBUILD23 name = "concrete_function",
25 "concrete_function.h",
48 ":concrete_function",
156 ":concrete_function",
184 "concrete_function.h",
/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.py203 for index, concrete_function in enumerate(concrete_functions, 1):
205 if concrete_function.structured_input_signature:
206 args, kwargs = concrete_function.structured_input_signature
207 elif concrete_function._arg_keywords: # pylint: disable=protected-access
210 args = concrete_function._arg_keywords # pylint: disable=protected-access
/external/tensorflow/tensorflow/python/keras/distribute/
Dsaved_model_save_load_test.py91 concrete_function = func.signatures[test_base._DEFAULT_FUNCTION_KEY]
92 for f in concrete_function.graph.as_graph_def().library.function:
/external/tensorflow/tensorflow/c/experimental/saved_model/
DREADME.md53 The files fulfilling 1. are named `*.cc` (eg: `concrete_function.cc`), while
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
DBUILD200 "//tensorflow/c/experimental/saved_model/core:concrete_function",
/external/tensorflow/tensorflow/lite/g3doc/convert/
Dindex.md112 ### Convert concrete functions <a name="concrete_function"></a>
Drnn.md37 # build a saved model. Here concrete_function is the exported function

12