Home
last modified time | relevance | path

Searched refs:saved_model_proto (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/cc/saved_model/
Dreader.cc31 Status ReadSavedModel(const string& export_dir, SavedModel* saved_model_proto) { in ReadSavedModel() argument
38 saved_model_proto); in ReadSavedModel()
44 saved_model_proto); in ReadSavedModel()
53 SavedModel* saved_model_proto, in FindMetaGraphDef() argument
57 for (MetaGraphDef& graph_def : *saved_model_proto->mutable_meta_graphs()) { in FindMetaGraphDef()
83 SavedModel saved_model_proto; in ReadMetaGraphDefFromSavedModel() local
84 TF_RETURN_IF_ERROR(ReadSavedModel(export_dir, &saved_model_proto)); in ReadMetaGraphDefFromSavedModel()
86 FindMetaGraphDef(tags, &saved_model_proto, meta_graph_def)); in ReadMetaGraphDefFromSavedModel()
Dbundle_v2.cc31 SavedModel* saved_model_proto) { in ReadSavedModelProto() argument
38 saved_model_proto); in ReadSavedModelProto()
44 saved_model_proto); in ReadSavedModelProto()
98 SavedModel saved_model_proto; in Load() local
99 TF_RETURN_IF_ERROR(ReadSavedModelProto(export_dir, &saved_model_proto)); in Load()
103 if (saved_model_proto.meta_graphs_size() != 1) { in Load()
108 "contains ", saved_model_proto.meta_graphs_size())); in Load()
111 std::move(*saved_model_proto.mutable_meta_graphs(0)); in Load()
/external/tensorflow/tensorflow/python/saved_model/
Dload.py122 def __init__(self, object_graph_proto, saved_model_proto, export_dir, argument
124 meta_graph = saved_model_proto.meta_graphs[0]
875 saved_model_proto, debug_info = (
878 if (len(saved_model_proto.meta_graphs) == 1 and
879 saved_model_proto.meta_graphs[0].HasField("object_graph_def")):
880 meta_graph_def = saved_model_proto.meta_graphs[0]
894 loader = loader_cls(object_graph_proto, saved_model_proto, export_dir,
Dsave_test.py200 saved_model_proto = loader_impl.parse_saved_model(save_dir)
203 saved_model_proto.meta_graphs[0].meta_info_def.tensorflow_version)
206 saved_model_proto.meta_graphs[0].meta_info_def.tensorflow_git_version)
/external/tensorflow/tensorflow/lite/python/
Dlite.py528 def _contains_function_with_implements_attr(self, saved_model_proto): argument
529 meta_graph = saved_model_proto.meta_graphs[0]
548 saved_model_proto, _ = (
556 not self._contains_function_with_implements_attr(saved_model_proto)):
562 self._saved_model_version = saved_model_proto.saved_model_schema_version
Dlite_v2_test.py810 saved_model_proto = parse_saved_model(saved_model_dir)
811 saved_model_proto.saved_model_schema_version = 0
815 writer.write(saved_model_proto.SerializeToString())
/external/tensorflow/tensorflow/python/compiler/tensorrt/
Dtrt_convert_test.py643 saved_model_proto = loader_impl.parse_saved_model(export_dir)
644 self.assertEqual(1, len(saved_model_proto.meta_graphs))
645 meta_graph = saved_model_proto.meta_graphs[0]