Home
last modified time | relevance | path

Searched refs:GraphKeys (Results 1 – 25 of 109) sorted by relevance

12345

/external/tensorflow/tensorflow/python/training/
Dtraining_util.py90 global_step_tensors = graph.get_collection(ops.GraphKeys.GLOBAL_STEP)
126 ops.GraphKeys.GLOBAL_STEP,
133 ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.GLOBAL_STEP
138 ops.GraphKeys.GLOBAL_STEP,
144 collections=[ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.GLOBAL_STEP])
Dsupervisor.py406 ready_op = self._get_first_op_from_collection(ops.GraphKeys.READY_OP)
409 ops.add_to_collection(ops.GraphKeys.READY_OP, ready_op)
415 ops.GraphKeys.READY_FOR_LOCAL_INIT_OP)
428 init_op = self._get_first_op_from_collection(ops.GraphKeys.INIT_OP)
431 ops.add_to_collection(ops.GraphKeys.INIT_OP, init_op)
446 ops.GraphKeys.LOCAL_INIT_OP)
454 ops.add_to_collection(ops.GraphKeys.LOCAL_INIT_OP, local_init_op)
465 saver = self._get_first_op_from_collection(ops.GraphKeys.SAVERS)
468 ops.add_to_collection(ops.GraphKeys.SAVERS, saver)
479 summary_op = self._get_first_op_from_collection(ops.GraphKeys.SUMMARY_OP)
[all …]
Dsession_manager_test.py76 collections=[ops.GraphKeys.LOCAL_VARIABLES])
254 collections=[ops.GraphKeys.LOCAL_VARIABLES],
313 collections=[ops.GraphKeys.LOCAL_VARIABLES],
346 collections=[ops.GraphKeys.LOCAL_VARIABLES],
383 collections=[ops.GraphKeys.LOCAL_VARIABLES],
417 collections=[ops.GraphKeys.LOCAL_VARIABLES],
448 collections=[ops.GraphKeys.LOCAL_VARIABLES],
468 collections=[ops.GraphKeys.LOCAL_VARIABLES],
486 collections=[ops.GraphKeys.LOCAL_VARIABLES],
491 collections=[ops.GraphKeys.LOCAL_VARIABLES],
[all …]
Dtraining_util_test.py34 self.assertEqual('%s:0' % ops.GraphKeys.GLOBAL_STEP, global_step.name)
45 name=ops.GraphKeys.GLOBAL_STEP)
58 name=ops.GraphKeys.GLOBAL_STEP)
82 name=ops.GraphKeys.GLOBAL_STEP)
Devaluation.py48 eval_steps = graph.get_collection(ops.GraphKeys.EVAL_STEP)
60 collections=[ops.GraphKeys.LOCAL_VARIABLES, ops.GraphKeys.EVAL_STEP])
/external/tensorflow/tensorflow/python/ops/
Dresources.py55 ops.add_to_collection(ops.GraphKeys.RESOURCES, resource)
57 ops.add_to_collection(ops.GraphKeys.LOCAL_RESOURCES, resource)
62 return ops.get_collection(ops.GraphKeys.RESOURCES)
67 return ops.get_collection(ops.GraphKeys.LOCAL_RESOURCES)
Dlogging_ops.py428 _Collect(val, collections, [ops.GraphKeys.SUMMARIES])
490 _Collect(val, collections, [ops.GraphKeys.SUMMARIES])
551 _Collect(val, collections, [ops.GraphKeys.SUMMARIES])
590 def merge_all_summaries(key=ops.GraphKeys.SUMMARIES):
624 summary_op = ops.get_collection(ops.GraphKeys.SUMMARY_OP)
633 ops.add_to_collection(ops.GraphKeys.SUMMARY_OP, summary_op)
669 _Collect(val, collections, [ops.GraphKeys.SUMMARIES])
Dtemplate.py320 ops.get_collection_ref(ops.GraphKeys.GLOBAL_VARIABLES))
322 ops.get_collection_ref(ops.GraphKeys.TRAINABLE_VARIABLES))
330 ops.GraphKeys.TRAINABLE_VARIABLES)
343 variables = ops.get_collection_ref(ops.GraphKeys.GLOBAL_VARIABLES)
430 return ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES,
447 return ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES,
456 return ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES,
/external/tensorflow/tensorflow/python/grappler/
Ddatasets_test.py53 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
78 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
114 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
127 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
154 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
258 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
287 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
324 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
359 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
388 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
[all …]
Dcluster_test.py39 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
56 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
74 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
94 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
121 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
147 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
Ditem_test.py55 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
68 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
90 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
118 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
Dmodel_analyzer_test.py39 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
60 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
/external/tensorflow/tensorflow/python/ops/losses/
Dutil_test.py38 ops.GraphKeys.REGULARIZATION_LOSSES, constant_op.constant(2.0))
40 ops.GraphKeys.REGULARIZATION_LOSSES, constant_op.constant(3.0))
47 ops.GraphKeys.REGULARIZATION_LOSSES, constant_op.constant(-1.0))
Dlosses_impl.py140 losses, weights=1.0, scope=None, loss_collection=ops.GraphKeys.LOSSES,
209 loss_collection=ops.GraphKeys.LOSSES,
263 loss_collection=ops.GraphKeys.LOSSES,
317 loss_collection=ops.GraphKeys.LOSSES,
367 loss_collection=ops.GraphKeys.LOSSES,
443 loss_collection=ops.GraphKeys.LOSSES,
501 loss_collection=ops.GraphKeys.LOSSES):
597 loss_collection=ops.GraphKeys.LOSSES,
650 loss_collection=ops.GraphKeys.LOSSES,
714 loss_collection=ops.GraphKeys.LOSSES,
[all …]
Dutil.py179 def add_loss(loss, loss_collection=ops.GraphKeys.LOSSES):
194 def get_losses(scope=None, loss_collection=ops.GraphKeys.LOSSES):
217 return ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES, scope)
/external/tensorflow/tensorflow/python/summary/
Dsummary.py83 _summary_op_util.collect(val, collections, [_ops.GraphKeys.SUMMARIES])
139 _summary_op_util.collect(val, collections, [_ops.GraphKeys.SUMMARIES])
180 _summary_op_util.collect(val, collections, [_ops.GraphKeys.SUMMARIES])
229 _summary_op_util.collect(val, collections, [_ops.GraphKeys.SUMMARIES])
326 _summary_op_util.collect(val, collections, [_ops.GraphKeys.SUMMARIES])
377 def merge_all(key=_ops.GraphKeys.SUMMARIES, scope=None, name=None):
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph.py53 _COMPAT_COLLECTION_LIST = [ops.GraphKeys.LOCAL_VARIABLES,
54 ops.GraphKeys.MODEL_VARIABLES,
55 ops.GraphKeys.METRIC_VARIABLES]
595 if clear_extraneous_savers and ctype == ops.GraphKeys.SAVERS:
814 if ops.GraphKeys.TRAINABLE_VARIABLES in meta_graph_def.collection_def:
816 (ops.GraphKeys.TRAINABLE_VARIABLES,
817 meta_graph_def.collection_def[ops.GraphKeys.TRAINABLE_VARIABLES]))
819 if key != ops.GraphKeys.TRAINABLE_VARIABLES:
841 if key == ops.GraphKeys.METRIC_VARIABLES:
843 from_proto = ops.get_from_proto_function(ops.GraphKeys.GLOBAL_VARIABLES)
[all …]
Dmeta_graph_test.py274 global_vars = graph1.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
275 trainable_vars = graph1.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
286 global_vars = graph2.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
287 trainable_vars = graph2.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
302 orig_meta_graph.collection_def[ops.GraphKeys.METRIC_VARIABLES].CopyFrom(
307 var_list = graph2.get_collection(ops.GraphKeys.METRIC_VARIABLES)
639 [ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.TRAINABLE_VARIABLES]
671 ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.TRAINABLE_VARIABLES
675 include_collection_keys=[ops.GraphKeys.GLOBAL_VARIABLES],
676 omit_collection_keys=[ops.GraphKeys.TRAINABLE_VARIABLES])
[all …]
/external/tensorflow/tensorflow/python/eager/
Dwrap_function.py77 collections = [ops.GraphKeys.GLOBAL_VARIABLES]
78 if v.trainable and ops.GraphKeys.TRAINABLE_VARIABLES not in collections:
79 collections = list(collections) + [ops.GraphKeys.TRAINABLE_VARIABLES]
162 ops.GraphKeys.GLOBAL_VARIABLES)
164 ops.GraphKeys.LOCAL_VARIABLES)
193 ops.GraphKeys.GLOBAL_VARIABLES)
196 ops.GraphKeys.LOCAL_VARIABLES, new_variable)
201 ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.LOCAL_VARIABLES
Dwrap_function_test.py209 ops.add_to_collection(ops.GraphKeys.LOSSES, v2 * constant_op.constant(3.))
215 len(f_wrapped.graph.get_collection(ops.GraphKeys.LOSSES)), 1)
217 ops.GraphKeys.TRAINABLE_VARIABLES)
225 ops.add_to_collection(ops.GraphKeys.LOSSES, v3 * constant_op.constant(3.))
231 len(g_wrapped.graph.get_collection(ops.GraphKeys.LOSSES)), 1)
233 ops.GraphKeys.TRAINABLE_VARIABLES)
238 self.assertIsNot(g_wrapped.graph.get_collection(ops.GraphKeys.LOSSES[0]),
239 f_wrapped.graph.get_collection(ops.GraphKeys.LOSSES)[0])
428 self.assertLen(graph.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES), 1)
/external/tensorflow/tensorflow/python/feature_column/
Ddense_features_v2_test.py391 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
398 [v.name for v in ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)])
429 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
432 [v.name for v in ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)])
462 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
481 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
484 [v.name for v in ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)])
Ddense_features_test.py444 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
451 [v.name for v in ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)])
482 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
485 [v.name for v in ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)])
515 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
534 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
537 [v.name for v in ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)])
/external/tensorflow/tensorflow/python/saved_model/
Dload_v1_in_v2.py116 ops.add_to_collection(ops.GraphKeys.TABLE_INITIALIZERS, initializer)
119 ops.GraphKeys.GLOBAL_VARIABLES):
238 ops.add_to_collection(ops.GraphKeys.TABLE_INITIALIZERS, local_init_op)
240 ops.GraphKeys.LOCAL_VARIABLES):
Dsaved_model_test.py79 ops.add_to_collection(ops.GraphKeys.ASSET_FILEPATHS, asset_file_tensor)
80 asset_collection = ops.get_collection(ops.GraphKeys.ASSET_FILEPATHS)
223 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
290 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
297 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
304 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
311 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
318 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
365 collection_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
375 collection_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
[all …]
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_experimental.py280 g.add_to_collection(ops.GraphKeys.GLOBAL_STEP, clone.optimizer.iterations)
291 g.get_collection_ref(ops.GraphKeys.UPDATE_OPS).extend(clone.state_updates)
343 local_vars = set(ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES))
354 ops.add_to_collection(ops.GraphKeys.LOCAL_VARIABLES, v)

12345