HomeSort by relevance Sort by last modified time
    Searched refs:collection_def (Results 1 - 24 of 24) sorted by null

  /external/tensorflow/tensorflow/contrib/session_bundle/
session_bundle.py 130 collection_def = meta_graph_def.collection_def
132 if constants.GRAPH_KEY in collection_def:
133 # Use serving graph_def in MetaGraphDef collection_def if exists
134 graph_def_any = collection_def[constants.GRAPH_KEY].any_list.value
152 if constants.INIT_OP_KEY in collection_def:
153 init_ops = collection_def[constants.INIT_OP_KEY].node_list.value
161 if constants.ASSETS_KEY in collection_def:
162 assets_any = collection_def[constants.ASSETS_KEY].any_list.value
exporter_test.py 143 collection_def = meta_graph_def.collection_def
146 graph_def_any = collection_def[constants.GRAPH_KEY].any_list.value
156 init_ops = collection_def[constants.INIT_OP_KEY].node_list.value
161 signatures_any = collection_def[constants.SIGNATURES_KEY].any_list.value
177 assets_any = collection_def[constants.ASSETS_KEY].any_list.value
session_bundle_test.py 105 collection_def = meta_graph_def.collection_def
107 signatures_any = collection_def[constants.SIGNATURES_KEY].any_list.value
143 collection_def = meta_graph_def.collection_def
145 signatures_any = collection_def[constants.SIGNATURES_KEY].any_list.value
signature.cc 51 const auto& collection_def = meta_graph_def.collection_def(); local
52 const auto it = collection_def.find(kSignaturesKey);
53 if (it == collection_def.end() || it->second.any_list().value_size() != 1) {
64 auto& collection_def = *(meta_graph_def->mutable_collection_def()); local
65 auto* any_list = collection_def[kSignaturesKey].mutable_any_list();
bundle_shim_test.py 288 collection_def = metagraph_def.collection_def
290 signatures = collection_def[constants.SIGNATURES_KEY].any_list.value[0]
299 metagraph_def.collection_def[constants.SIGNATURES_KEY].any_list.value[
308 metagraph_def.collection_def[constants.SIGNATURES_KEY].any_list.value[
335 collection_def = meta_graph_def.collection_def
337 signatures_any = collection_def[constants.SIGNATURES_KEY].any_list.value
bundle_shim.py 168 collection_def = metagraph_def.collection_def
170 signatures = collection_def[legacy_constants.SIGNATURES_KEY].any_list.value[0]
signature_test.cc 497 auto& collection_def = *(meta_graph_def.mutable_collection_def()); local
499 collection_def[kSignaturesKey].mutable_any_list()->mutable_value()->Add();
513 auto& collection_def = *(meta_graph_def.mutable_collection_def()); local
515 collection_def[kSignaturesKey].mutable_any_list()->mutable_value()->Add();
528 auto& collection_def = *(meta_graph_def.mutable_collection_def()); local
530 collection_def[kSignaturesKey].mutable_any_list()->mutable_value()->Add();
548 // Check that we only have one 'Signatures' entry in the collection_def map.
552 auto& collection_def = *(meta_graph_def.mutable_collection_def()); local
554 collection_def[kSignaturesKey].mutable_any_list()->mutable_value()->Add();
564 collection_def[kSignaturesKey].mutable_any_list()->mutable_value()->Add()
    [all...]
session_bundle.cc 174 const auto& collection_def_map = bundle->meta_graph_def.collection_def();
175 const auto graph_it = bundle->meta_graph_def.collection_def().find(kGraphKey);
178 // Use serving graph_def in MetaGraphDef collection_def.
  /external/tensorflow/tensorflow/python/saved_model/
loader_impl.py 94 collection_def = meta_graph_def_to_load.collection_def
97 if constants.ASSETS_KEY in collection_def:
102 assets_any_proto = collection_def[constants.ASSETS_KEY].any_list.value
126 collection_def = meta_graph_def_to_load.collection_def
128 if constants.MAIN_OP_KEY in collection_def:
129 main_ops = collection_def[constants.MAIN_OP_KEY].node_list.value
149 collection_def = meta_graph_def_to_load.collection_def
    [all...]
saved_model_test.py 583 self._validate_asset_collection(export_dir, foo_graph.collection_def,
720 self._validate_asset_collection(export_dir, foo_graph.collection_def,
727 self._validate_asset_collection(export_dir, bar_graph.collection_def,
    [all...]
  /external/tensorflow/tensorflow/core/grappler/inputs/
file_input_yielder.cc 86 if (metagraph.collection_def().count("train_op") == 0 ||
87 !metagraph.collection_def().at("train_op").has_node_list() ||
88 metagraph.collection_def().at("train_op").node_list().value_size() == 0) {
96 metagraph.collection_def().at("train_op").node_list().value()) {
  /external/tensorflow/tensorflow/core/grappler/
grappler_item_builder.cc 181 if (meta_graph.collection_def().count("train_op") > 0) {
182 const CollectionDef& nodes = meta_graph.collection_def().at("train_op");
210 if (meta_graph.collection_def().count(var_collection) == 0) {
213 const CollectionDef& vars = meta_graph.collection_def().at(var_collection);
223 if (meta_graph.collection_def().count("table_initializer") > 0) {
225 meta_graph.collection_def().at("table_initializer");
247 if (meta_graph.collection_def().count("saved_model_assets") > 0) {
249 meta_graph.collection_def().at("saved_model_assets");
269 } else if (meta_graph.collection_def().count("asset_filepaths") > 0) {
271 meta_graph.collection_def().at("asset_filepaths")
    [all...]
  /external/tensorflow/tensorflow/contrib/meta_graph_transform/
meta_graph_transform.py 463 base_collection = base_meta_graph_def.collection_def[collection_name]
464 collection = meta_graph_def.collection_def[collection_name]
581 if collection_key not in meta_graph_def.collection_def:
583 collection = meta_graph_def.collection_def[collection_key]
597 if collection_key not in meta_graph_def.collection_def:
599 collection = meta_graph_def.collection_def[collection_key]
609 if collection_key not in meta_graph_def.collection_def:
611 collection = meta_graph_def.collection_def[collection_key]
700 meta_graph_def.collection_def[
754 for collection_name in base_meta_graph_def.collection_def
    [all...]
meta_graph_transform_test.py 351 meta_graph_def.collection_def[
361 self.assertEqual(meta_graph_def.collection_def[
388 base_meta_graph_def.collection_def[
391 base_meta_graph_def.collection_def[
453 base_meta_graph_def.collection_def[collection_name].node_list.value.extend(
461 collection = meta_graph_def.collection_def[collection_name]
469 base_meta_graph_def.collection_def[collection_name].int64_list.value[:] = (
477 collection = meta_graph_def.collection_def[collection_name]
486 base_meta_graph_def.collection_def[collection_name].any_list.value.extend([
500 collection = meta_graph_def.collection_def[collection_name
    [all...]
  /external/tensorflow/tensorflow/cc/saved_model/
loader.cc 126 const auto& collection_def_map = meta_graph_def.collection_def();
139 const auto& collection_def_map = meta_graph_def.collection_def();
197 const auto& collection_def_map = meta_graph_def.collection_def();
218 const auto& collection_def_map = meta_graph_def.collection_def();
  /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
export_test.py 59 collection_def = meta_graph_def.collection_def
61 signatures_any = collection_def['serving_signatures'].any_list.value
  /external/tensorflow/tensorflow/python/grappler/
cost_analyzer_tool.py 49 metagraph.collection_def["train_op"].CopyFrom(fetch_collection)
item.i 76 if (meta_graph.collection_def().count("train_op") == 0) {
  /external/tensorflow/tensorflow/python/training/
saver_test.py     [all...]
  /external/tensorflow/tensorflow/python/framework/
meta_graph.py 415 col_def = meta_graph_def.collection_def[key]
445 if key in meta_graph_def.collection_def:
446 del meta_graph_def.collection_def[key]
707 for key, col_def in meta_graph_def.collection_def.items():
745 for key, col_def in sorted(meta_graph_def.collection_def.items()):
    [all...]
test_util.py 145 tester.assertEqual(set(a.collection_def), set(b.collection_def))
146 collection_keys = a.collection_def.keys()
148 a_value = a.collection_def[k]
149 b_value = b.collection_def[k]
166 a.ClearField("collection_def")
167 b.ClearField("collection_def")
    [all...]
meta_graph_test.py 471 del a.collection_def["unbound_inputs"]
472 del b.collection_def["unbound_inputs"]
    [all...]
  /external/tensorflow/tensorflow/python/framework/testdata/
metrics_export_meta_graph.pb 1542 collection_def {
1551 collection_def {
  /external/tensorflow/tensorflow/contrib/training/python/training/
batch_sequences_with_states_test.py 142 qr_saved = meta_graph_def.collection_def[ops.GraphKeys.QUEUE_RUNNERS]

Completed in 243 milliseconds