Home
last modified time | relevance | path

Searched refs:concrete_functions (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/python/saved_model/
Dfunction_deserialization.py164 concrete_functions): argument
166 concrete_function = concrete_functions[
190 def __init__(self, python_function, name, function_spec, concrete_functions): argument
195 self.concrete_functions = concrete_functions
217 return self.concrete_functions
225 def recreate_function(saved_function, concrete_functions): argument
258 if not saved_function.concrete_functions:
268 for function_name in saved_function.concrete_functions:
269 function = concrete_functions[function_name]
279 for index, function_name in enumerate(saved_function.concrete_functions):
[all …]
Dsave.py211 self.concrete_functions = []
251 concrete_functions = (
254 concrete_functions = [function]
255 if not concrete_functions:
257 for concrete_function in concrete_functions:
259 self.concrete_functions.append(concrete_function)
354 for concrete_function in self.concrete_functions:
386 self.concrete_functions = [
387 self._wrapped_functions.get(x, x) for x in self.concrete_functions
702 for concrete_function in saveable_view.concrete_functions:
[all …]
Dload.py313 for name, proto in sorted(self._proto.concrete_functions.items()):
334 concrete_functions = sorted(self._proto.concrete_functions.items())
335 for name, proto in concrete_functions:
Dfunction_serialization.py114 proto.concrete_functions.append(
Dload_test.py197 imported_concrete, = imported.f.concrete_functions
408concrete_functions = root.f._list_all_concrete_functions_for_serialization() # pylint: disable=pr…
409 self.assertLen(concrete_functions, 4)
439concrete_functions = root.f._list_all_concrete_functions_for_serialization() # pylint: disable=pr…
440 self.assertLen(concrete_functions, 3)
950concrete_functions = root.f._list_all_concrete_functions_for_serialization() # pylint: disable=pr…
951 self.assertLen(concrete_functions, 1)
/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Dpartially_revived_objects.cc346 revived->concrete_functions.Insert(std::move(out), in InitializeCreateResourceFunctions()
359 for (const auto& id_and_func : objects.concrete_functions) { in InitializeAllFunctions()
363 if (revived->concrete_functions.Find(node_id)) { in InitializeAllFunctions()
373 revived->concrete_functions.Insert(std::move(out), node_id); in InitializeAllFunctions()
402 revived->concrete_functions.Find(create_resource_fn_node); in CreateAllResourceHandles()
443 create_resource = revived->concrete_functions.Find( in BuildResources()
454 initialize = revived->concrete_functions.Find( in BuildResources()
465 destroy_resource = revived->concrete_functions.Find( in BuildResources()
Dpartially_revived_objects.h51 gtl::FlatMap<int, TFConcreteFunctionRevivalState> concrete_functions; member
Drevived_objects.h85 RevivedObjectContainer<TFConcreteFunction> concrete_functions; member
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dsaved_model_utils.cc446 const std::string& function_name = saved_function.concrete_functions(0); in PartiallyReviveSavedModelObjects()
449 metagraph.object_graph_def().concrete_functions().at(function_name); in PartiallyReviveSavedModelObjects()
469 objects->concrete_functions[i] = std::move(func_revival_state); in PartiallyReviveSavedModelObjects()
478 metagraph.object_graph_def().concrete_functions().at(function_name); in PartiallyReviveSavedModelObjects()
494 objects->concrete_functions[i] = std::move(func_revival_state); in PartiallyReviveSavedModelObjects()
516 resource.create_resource = &objects->concrete_functions[child_node_id]; in PartiallyReviveSavedModelObjects()
520 resource.initialize = &objects->concrete_functions[child_node_id]; in PartiallyReviveSavedModelObjects()
524 resource.destroy_resource = &objects->concrete_functions[child_node_id]; in PartiallyReviveSavedModelObjects()
Dtf_saved_model_api.cc183 *function = revived_objects_.concrete_functions.Find(*node); in GetFunction()
280 revived_objects.concrete_functions.Insert(std::move(concrete_function)); in Load()
/external/tensorflow/tensorflow/python/eager/
Ddef_function.py1146 concrete_functions = []
1149 concrete_functions.extend(
1152 concrete_functions.extend(
1155 return concrete_functions
1163 concrete_functions = self._list_all_concrete_functions()
1165 for concrete_function in concrete_functions:
1180 concrete_functions = []
1182 concrete_functions.append(self.get_concrete_function(*args, **kwargs))
1183 return concrete_functions
Ddef_function_test.py419 concrete_functions = f._list_all_concrete_functions_for_serialization()
420 for concrete_function in concrete_functions:
Dfunction_test.py784 concrete_functions = []
788 concrete_functions.append(cf)
794 self.assertLen(set(concrete_functions), 1)
/external/tensorflow/tensorflow/python/tools/
Dsaved_model_cli.py196 concrete_functions = []
198 concrete_functions.append(function)
200 concrete_functions.extend(
202 concrete_functions = sorted(concrete_functions, key=lambda x: x.name)
203 for index, concrete_function in enumerate(concrete_functions, 1):
/external/tensorflow/tensorflow/python/keras/saving/saved_model/
Dload.py768 concrete_functions = (
769 self._proto.nodes[call_fn_id].function.concrete_functions)
770 if not concrete_functions:
772 call_fn_name = concrete_functions[0]
773 call_fn_proto = self._proto.concrete_functions[call_fn_name]
1125 spec = fn.concrete_functions[0].structured_input_signature[0][0]
1126 for concrete in fn.concrete_functions[1:]:
/external/llvm-project/llvm/unittests/Analysis/Inputs/ir2native_x86_64_model/
Dsaved_model.pbtxt9254 concrete_functions: "__inference_sequential_layer_call_fn_6629"
9255 concrete_functions: "__inference_sequential_layer_call_fn_6733"
9256 concrete_functions: "__inference_sequential_layer_call_fn_6720"
9257 concrete_functions: "__inference_sequential_layer_call_fn_6656"
9342 concrete_functions: "__inference__wrapped_model_6528"
9420 concrete_functions: "__inference_sequential_layer_call_and_return_conditional_losses_6689"
9421 concrete_functions: "__inference_sequential_layer_call_and_return_conditional_losses_6587"
9422 concrete_functions: "__inference_sequential_layer_call_and_return_conditional_losses_6707"
9423 concrete_functions: "__inference_sequential_layer_call_and_return_conditional_losses_6601"
9508 concrete_functions: "__inference_dense_layer_call_fn_6754"
[all …]
/external/tensorflow/tensorflow/core/protobuf/
Dsaved_object_graph.proto33 map<string, SavedConcreteFunction> concrete_functions = 2; field
101 repeated string concrete_functions = 1; field
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc3109 tf_name_to_mlir_name.find(function.concrete_functions(0))->second); in CreateSavedModelIR()
3137 object_graph.concrete_functions().at(function.concrete_functions(0)); in CreateSavedModelIR()
3163 function.concrete_functions(0), "' (", input_index_paths.size(), in CreateSavedModelIR()
3189 function.concrete_functions(0), "' (", output_index_paths.size(), in CreateSavedModelIR()
/external/llvm-project/llvm/lib/Analysis/models/inliner/
Dsaved_model.pbtxt28450 concrete_functions: "__inference_polymorphic_action_fn_4619080"
28451 concrete_functions: "__inference_polymorphic_action_fn_946"
28526 concrete_functions: "__inference_function_722"
31717 concrete_functions {
31742 concrete_functions {
31763 concrete_functions {
32045 concrete_functions {
32327 concrete_functions {
32578 concrete_functions {
32599 concrete_functions {