Searched refs:revived_objects (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | tf_saved_model_api.cc | 111 const RevivedObjects& revived_objects, in RestoreCheckpoint() argument
117 [&revived_objects, &directory, context, bundle]( in RestoreCheckpoint()
131 Variable* variable = revived_objects.variables.at(node).get(); in RestoreCheckpoint()
231 RevivedObjects revived_objects) in TFSavedModelAPI() argument
234 revived_objects_(std::move(revived_objects)) {} in TFSavedModelAPI()
263 RevivedObjects revived_objects; in Load() local
265 context, bundle.saved_object_graph(), &revived_objects)); in Load()
280 revived_objects.concrete_functions.Insert(std::move(concrete_function)); in Load()
284 RestoreCheckpoint(&bundle, revived_objects, directory, context)); in Load()
286 TF_RETURN_IF_ERROR(InitializeAllResources(revived_objects)); in Load()
[all …]
|
| D | tf_saved_model_api.h | 75 RevivedObjects revived_objects);
|
| D | BUILD | 166 "//tensorflow/c/experimental/saved_model/core/revived_types:revived_objects",
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/ |
| D | BUILD | 85 ":revived_objects",
137 name = "revived_objects",
139 "revived_objects.h",
|