Home
last modified time | relevance | path

Searched refs:obj_graph (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/c/experimental/saved_model/core/revived_types/
Dpartially_revived_objects.cc80 Status TensorHandleFromNode(int node_id, const SavedObjectGraph& obj_graph, in TensorHandleFromNode() argument
83 const SavedObject& node = obj_graph.nodes(node_id); in TensorHandleFromNode()
281 const SavedObjectGraph& obj_graph, in CreateConcreteFunction() argument
289 TF_RETURN_IF_ERROR(TensorHandleFromNode(capture_node_id, obj_graph, objects, in CreateConcreteFunction()
304 const SavedObjectGraph& obj_graph, const PartiallyRevivedObjects& objects, in CreateSignatureDefFunction() argument
311 TF_RETURN_IF_ERROR(TensorHandleFromNode(capture_node_id, obj_graph, objects, in CreateSignatureDefFunction()
328 const SavedObjectGraph& obj_graph, in InitializeCreateResourceFunctions() argument
345 obj_graph, objects, &out)); in InitializeCreateResourceFunctions()
353 const SavedObjectGraph& obj_graph, in InitializeAllFunctions() argument
372 CreateConcreteFunction(ctx, func, obj_graph, objects, &out)); in InitializeAllFunctions()
[all …]
Dpartially_revived_objects.h57 const SavedObjectGraph& obj_graph, RevivedObjects* revived);
/external/tensorflow/tensorflow/c/experimental/saved_model/core/
Dsaved_model_utils.cc503 const SavedObjectGraph& obj_graph = metagraph.object_graph_def(); in PartiallyReviveSavedModelObjects() local
504 const SavedObject& node = obj_graph.nodes(node_id); in PartiallyReviveSavedModelObjects()
514 obj_graph.nodes(child.node_id()).kind_case() == in PartiallyReviveSavedModelObjects()
518 obj_graph.nodes(child.node_id()).kind_case() == in PartiallyReviveSavedModelObjects()
522 obj_graph.nodes(child.node_id()).kind_case() == in PartiallyReviveSavedModelObjects()