OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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