Searched refs:saved_model_proto (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/cc/saved_model/ |
| D | reader.cc | 30 Status ReadSavedModel(const string& export_dir, SavedModel* saved_model_proto) { in ReadSavedModel() argument
37 saved_model_proto); in ReadSavedModel()
43 saved_model_proto); in ReadSavedModel()
52 SavedModel* saved_model_proto, in FindMetaGraphDef() argument
56 for (MetaGraphDef& graph_def : *saved_model_proto->mutable_meta_graphs()) { in FindMetaGraphDef()
82 SavedModel saved_model_proto; in ReadMetaGraphDefFromSavedModel() local
83 TF_RETURN_IF_ERROR(ReadSavedModel(export_dir, &saved_model_proto)); in ReadMetaGraphDefFromSavedModel()
85 FindMetaGraphDef(tags, &saved_model_proto, meta_graph_def)); in ReadMetaGraphDefFromSavedModel()
|
| D | bundle_v2.cc | 31 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/ |
| D | load.py | 106 def __init__(self, object_graph_proto, saved_model_proto, export_dir): argument
107 meta_graph = saved_model_proto.meta_graphs[0]
582 saved_model_proto, debug_info = (
585 if (len(saved_model_proto.meta_graphs) == 1 and
586 saved_model_proto.meta_graphs[0].HasField("object_graph_def")):
587 meta_graph_def = saved_model_proto.meta_graphs[0]
598 saved_model_proto,
|
| D | save_test.py | 187 saved_model_proto = loader_impl.parse_saved_model(save_dir)
190 saved_model_proto.meta_graphs[0].meta_info_def.tensorflow_version)
193 saved_model_proto.meta_graphs[0].meta_info_def.tensorflow_git_version)
|
| /external/tensorflow/tensorflow/python/compiler/tensorrt/ |
| D | trt_convert_test.py | 689 saved_model_proto = loader_impl.parse_saved_model(export_dir)
690 self.assertEqual(1, len(saved_model_proto.meta_graphs))
691 meta_graph = saved_model_proto.meta_graphs[0]
|