Home
last modified time | relevance | path

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

12345

/external/tensorflow/tensorflow/python/training/
Dtraining_util_test.py28 self.assertEqual('%s:0' % ops.GraphKeys.GLOBAL_STEP, global_step.name)
39 name=ops.GraphKeys.GLOBAL_STEP,
40 collections=[ops.GraphKeys.GLOBAL_STEP])
53 name=ops.GraphKeys.GLOBAL_STEP,
54 collections=[ops.GraphKeys.GLOBAL_STEP])
78 name=ops.GraphKeys.GLOBAL_STEP,
79 collections=[ops.GraphKeys.GLOBAL_STEP])
Dtraining_util.py145 global_step_tensors = graph.get_collection(ops.GraphKeys.GLOBAL_STEP)
234 ops.GraphKeys.GLOBAL_STEP,
241 ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.GLOBAL_STEP
246 ops.GraphKeys.GLOBAL_STEP,
252 collections=[ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.GLOBAL_STEP])
Dsupervisor.py402 ready_op = self._get_first_op_from_collection(ops.GraphKeys.READY_OP)
405 ops.add_to_collection(ops.GraphKeys.READY_OP, ready_op)
411 ops.GraphKeys.READY_FOR_LOCAL_INIT_OP)
424 init_op = self._get_first_op_from_collection(ops.GraphKeys.INIT_OP)
427 ops.add_to_collection(ops.GraphKeys.INIT_OP, init_op)
442 ops.GraphKeys.LOCAL_INIT_OP)
450 ops.add_to_collection(ops.GraphKeys.LOCAL_INIT_OP, local_init_op)
461 saver = self._get_first_op_from_collection(ops.GraphKeys.SAVERS)
464 ops.add_to_collection(ops.GraphKeys.SAVERS, saver)
475 summary_op = self._get_first_op_from_collection(ops.GraphKeys.SUMMARY_OP)
[all …]
Dsession_manager_test.py77 collections=[ops.GraphKeys.LOCAL_VARIABLES])
251 collections=[ops.GraphKeys.LOCAL_VARIABLES],
309 collections=[ops.GraphKeys.LOCAL_VARIABLES],
341 collections=[ops.GraphKeys.LOCAL_VARIABLES],
377 collections=[ops.GraphKeys.LOCAL_VARIABLES],
410 collections=[ops.GraphKeys.LOCAL_VARIABLES],
441 collections=[ops.GraphKeys.LOCAL_VARIABLES],
461 collections=[ops.GraphKeys.LOCAL_VARIABLES],
478 collections=[ops.GraphKeys.LOCAL_VARIABLES],
483 collections=[ops.GraphKeys.LOCAL_VARIABLES],
[all …]
Devaluation.py44 eval_steps = graph.get_collection(ops.GraphKeys.EVAL_STEP)
56 collections=[ops.GraphKeys.LOCAL_VARIABLES, ops.GraphKeys.EVAL_STEP])
/external/tensorflow/tensorflow/python/ops/
Dresources.py51 ops.add_to_collection(ops.GraphKeys.RESOURCES, resource)
53 ops.add_to_collection(ops.GraphKeys.LOCAL_RESOURCES, resource)
58 return ops.get_collection(ops.GraphKeys.RESOURCES)
63 return ops.get_collection(ops.GraphKeys.LOCAL_RESOURCES)
Dlogging_ops.py445 _Collect(val, collections, [ops.GraphKeys.SUMMARIES])
507 _Collect(val, collections, [ops.GraphKeys.SUMMARIES])
568 _Collect(val, collections, [ops.GraphKeys.SUMMARIES])
607 def merge_all_summaries(key=ops.GraphKeys.SUMMARIES):
641 summary_op = ops.get_collection(ops.GraphKeys.SUMMARY_OP)
650 ops.add_to_collection(ops.GraphKeys.SUMMARY_OP, summary_op)
686 _Collect(val, collections, [ops.GraphKeys.SUMMARIES])
Dtemplate.py334 ops.get_collection_ref(ops.GraphKeys.GLOBAL_VARIABLES))
336 ops.get_collection_ref(ops.GraphKeys.TRAINABLE_VARIABLES))
344 ops.GraphKeys.TRAINABLE_VARIABLES)
357 variables = ops.get_collection_ref(ops.GraphKeys.GLOBAL_VARIABLES)
444 return ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES,
461 return ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES,
470 return ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES,
/external/tensorflow/tensorflow/python/grappler/
Ddatasets_test.py49 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
74 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
110 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
123 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
150 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
254 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
283 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
320 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
355 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
384 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
[all …]
Dcluster_test.py35 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
52 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
70 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
90 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
117 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
143 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
Ditem_test.py51 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
64 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
86 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
114 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
Dmodel_analyzer_test.py35 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
56 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
Dcost_analyzer_test.py44 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
69 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
149 train_op = ops.get_collection_ref(ops.GraphKeys.TRAIN_OP)
/external/tensorflow/tensorflow/python/ops/losses/
Dutil_test.py34 ops.GraphKeys.REGULARIZATION_LOSSES, constant_op.constant(2.0))
36 ops.GraphKeys.REGULARIZATION_LOSSES, constant_op.constant(3.0))
43 ops.GraphKeys.REGULARIZATION_LOSSES, constant_op.constant(-1.0))
Dlosses_impl.py139 losses, weights=1.0, scope=None, loss_collection=ops.GraphKeys.LOSSES,
219 loss_collection=ops.GraphKeys.LOSSES,
274 loss_collection=ops.GraphKeys.LOSSES,
329 loss_collection=ops.GraphKeys.LOSSES,
380 loss_collection=ops.GraphKeys.LOSSES,
457 loss_collection=ops.GraphKeys.LOSSES,
516 loss_collection=ops.GraphKeys.LOSSES):
624 loss_collection=ops.GraphKeys.LOSSES,
774 loss_collection=ops.GraphKeys.LOSSES,
840 loss_collection=ops.GraphKeys.LOSSES,
[all …]
Dutil.py175 def add_loss(loss, loss_collection=ops.GraphKeys.LOSSES):
190 def get_losses(scope=None, loss_collection=ops.GraphKeys.LOSSES):
213 return ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES, scope)
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph.py48 _COMPAT_COLLECTION_LIST = [ops.GraphKeys.LOCAL_VARIABLES,
49 ops.GraphKeys.MODEL_VARIABLES,
50 ops.GraphKeys.METRIC_VARIABLES]
603 if clear_extraneous_savers and ctype == ops.GraphKeys.SAVERS:
822 if ops.GraphKeys.TRAINABLE_VARIABLES in meta_graph_def.collection_def:
824 (ops.GraphKeys.TRAINABLE_VARIABLES,
825 meta_graph_def.collection_def[ops.GraphKeys.TRAINABLE_VARIABLES]))
827 if key != ops.GraphKeys.TRAINABLE_VARIABLES:
849 if key == ops.GraphKeys.METRIC_VARIABLES:
851 from_proto = ops.get_from_proto_function(ops.GraphKeys.GLOBAL_VARIABLES)
[all …]
Dmeta_graph_test.py293 global_vars = graph1.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
294 trainable_vars = graph1.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
305 global_vars = graph2.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
306 trainable_vars = graph2.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
321 orig_meta_graph.collection_def[ops.GraphKeys.METRIC_VARIABLES].CopyFrom(
326 var_list = graph2.get_collection(ops.GraphKeys.METRIC_VARIABLES)
658 [ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.TRAINABLE_VARIABLES]
690 ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.TRAINABLE_VARIABLES
694 include_collection_keys=[ops.GraphKeys.GLOBAL_VARIABLES],
695 omit_collection_keys=[ops.GraphKeys.TRAINABLE_VARIABLES])
[all …]
/external/tensorflow/tensorflow/python/summary/
Dsummary.py136 _summary_op_util.collect(val, collections, [_ops.GraphKeys.SUMMARIES])
252 _summary_op_util.collect(val, collections, [_ops.GraphKeys.SUMMARIES])
344 _summary_op_util.collect(val, collections, [_ops.GraphKeys.SUMMARIES])
465 _summary_op_util.collect(val, collections, [_ops.GraphKeys.SUMMARIES])
610 _summary_op_util.collect(val, collections, [_ops.GraphKeys.SUMMARIES])
697 def merge_all(key=_ops.GraphKeys.SUMMARIES, scope=None, name=None):
/external/tensorflow/tensorflow/python/eager/
Dwrap_function.py74 collections = [ops.GraphKeys.GLOBAL_VARIABLES]
75 if v.trainable and ops.GraphKeys.TRAINABLE_VARIABLES not in collections:
76 collections = list(collections) + [ops.GraphKeys.TRAINABLE_VARIABLES]
160 ops.GraphKeys.GLOBAL_VARIABLES)
162 ops.GraphKeys.LOCAL_VARIABLES)
191 ops.GraphKeys.GLOBAL_VARIABLES)
194 ops.GraphKeys.LOCAL_VARIABLES, new_variable)
199 ops.GraphKeys.GLOBAL_VARIABLES, ops.GraphKeys.LOCAL_VARIABLES
Dwrap_function_test.py205 ops.add_to_collection(ops.GraphKeys.LOSSES, v2 * constant_op.constant(3.))
211 len(f_wrapped.graph.get_collection(ops.GraphKeys.LOSSES)), 1)
213 ops.GraphKeys.TRAINABLE_VARIABLES)
221 ops.add_to_collection(ops.GraphKeys.LOSSES, v3 * constant_op.constant(3.))
227 len(g_wrapped.graph.get_collection(ops.GraphKeys.LOSSES)), 1)
229 ops.GraphKeys.TRAINABLE_VARIABLES)
234 self.assertIsNot(g_wrapped.graph.get_collection(ops.GraphKeys.LOSSES[0]),
235 f_wrapped.graph.get_collection(ops.GraphKeys.LOSSES)[0])
424 self.assertLen(graph.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES), 1)
/external/tensorflow/tensorflow/python/distribute/
Ddistribute_utils.py296 var_collections = [ops.GraphKeys.GLOBAL_VARIABLES]
344 var_collections.append(ops.GraphKeys.TRAINABLE_VARIABLES)
345 l = g.get_collection_ref(ops.GraphKeys.TRAINABLE_VARIABLES)
353 elif ops.GraphKeys.GLOBAL_STEP in var_collections:
354 ops.add_to_collections(ops.GraphKeys.GLOBAL_STEP, result)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
Dhash_table_asset_v1.py58 tf.add_to_collection(tf.GraphKeys.ASSET_FILEPATHS, vocab_file_tensor)
71 }, tf.tables_initializer(), tf.get_collection(tf.GraphKeys.ASSET_FILEPATHS)
/external/tensorflow/tensorflow/python/keras/saving/
Dsaved_model_experimental.py279 g.add_to_collection(ops.GraphKeys.GLOBAL_STEP, clone.optimizer.iterations)
290 g.get_collection_ref(ops.GraphKeys.UPDATE_OPS).extend(clone.state_updates)
342 local_vars = set(ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES))
353 ops.add_to_collection(ops.GraphKeys.LOCAL_VARIABLES, v)
/external/tensorflow/tensorflow/python/kernel_tests/variables/
Dvariable_scope_test.py261 "v3", [], collections=[ops.GraphKeys.CONCATENATED_VARIABLES])
263 ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES),
266 ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES),
269 ops.get_collection(ops.GraphKeys.CONCATENATED_VARIABLES), [concat])
275 self.assertFalse(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES))
276 self.assertFalse(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES))
389 losses = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
405 losses = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
416 losses = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
1141 ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES))
[all …]

12345