Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/contrib/session_bundle/
Dbundle_shim.py154 def _convert_signatures_to_signature_defs(metagraph_def): argument
168 collection_def = metagraph_def.collection_def
210 metagraph_def = meta_graph.read_meta_graph_file(meta_graph_filename)
211 if metagraph_def.signature_def:
217 _convert_signatures_to_signature_defs(metagraph_def))
219 metagraph_def.signature_def[
226 metagraph_def.signature_def[signature_def_key].CopyFrom(named_signature_def)
233 sess, metagraph_def = session_bundle.load_session_bundle_from_path(
234 export_dir, target, config, meta_graph_def=metagraph_def)
235 return sess, metagraph_def
[all …]
Dbundle_shim_test.py265 metagraph_def = meta_graph.read_meta_graph_file(meta_graph_filename)
267 bundle_shim._convert_signatures_to_signature_defs(metagraph_def))
288 collection_def = metagraph_def.collection_def
299 metagraph_def.collection_def[constants.SIGNATURES_KEY].any_list.value[
302 bundle_shim._convert_signatures_to_signature_defs(metagraph_def))
308 metagraph_def.collection_def[constants.SIGNATURES_KEY].any_list.value[
311 bundle_shim._convert_signatures_to_signature_defs(metagraph_def))
/external/tensorflow/tensorflow/java/src/main/native/
Dsaved_model_bundle_jni.cc66 TF_Buffer* metagraph_def = TF_NewBuffer(); in Java_org_tensorflow_SavedModelBundle_load() local
69 metagraph_def, status); in Java_org_tensorflow_SavedModelBundle_load()
86 if (metagraph_def->length > std::numeric_limits<jint>::max()) { in Java_org_tensorflow_SavedModelBundle_load()
92 jint jmetagraph_len = static_cast<jint>(metagraph_def->length); in Java_org_tensorflow_SavedModelBundle_load()
95 static_cast<const jbyte*>(metagraph_def->data)); in Java_org_tensorflow_SavedModelBundle_load()
116 TF_DeleteBuffer(metagraph_def); in Java_org_tensorflow_SavedModelBundle_load()
/external/tensorflow/tensorflow/contrib/predictor/
Dsaved_model_predictor.py67 metagraph_def = get_meta_graph_def(export_dir, tags)
70 signature_def = metagraph_def.signature_def[signature_def_key]
75 signature_def = metagraph_def.signature_def[formatted_key]
80 signature_def_key, list(metagraph_def.signature_def), e))
/external/tensorflow/tensorflow/python/summary/
Dsummary_test.py228 metagraph_def, _ = meta_graph.export_scoped_meta_graph(export_scope='outer')
231 meta_graph.import_scoped_meta_graph(metagraph_def, graph=g,
/external/tensorflow/tensorflow/c/
Dc_api_test.cc1366 tensorflow::MetaGraphDef metagraph_def; in TEST() local
1367 metagraph_def.ParseFromArray(metagraph->data, metagraph->length); in TEST()
1374 const auto signature_def_map = metagraph_def.signature_def(); in TEST()
/external/tensorflow/
DRELEASE.md143 * Replace any calls to `tf.contrib.get_signature_def_by_key(metagraph_def,