Home
last modified time | relevance | path

Searched refs:get_collection (Results 1 – 25 of 84) sorted by relevance

1234

/external/tensorflow/tensorflow/python/eager/
Dfunction_defun_collection_test.py51 x_const = constant_op.constant(ops.get_collection('x')[0])
52 y_const = constant_op.constant(ops.get_collection('y')[0])
58 self.assertEquals(ops.get_collection('x'), [2])
59 self.assertEquals(ops.get_collection('y'), [5])
60 self.assertEquals(ops.get_collection('z'), [])
80 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)), 1)
96 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)), 1)
Dwrap_function_test.py215 len(f_wrapped.graph.get_collection(ops.GraphKeys.LOSSES)), 1)
216 f_var_collection = f_wrapped.graph.get_collection(
231 len(g_wrapped.graph.get_collection(ops.GraphKeys.LOSSES)), 1)
232 g_var_collection = g_wrapped.graph.get_collection(
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)
429 self.assertLen(graph.get_collection('a'), 2)
430 self.assertLen(graph.get_collection('b'), 1)
/external/tensorflow/tensorflow/python/saved_model/
Dsaved_model_test.py80 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)
440 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
[all …]
Dsimple_save.py88 assets_collection=ops.get_collection(ops.GraphKeys.ASSET_FILEPATHS),
/external/tensorflow/tensorflow/python/summary/
Dplugin_asset.py68 container = graph.get_collection(name)
103 for name in graph.get_collection(_PLUGIN_ASSET_PREFIX):
104 collection = graph.get_collection(_PLUGIN_ASSET_PREFIX + name)
/external/tensorflow/tensorflow/python/ops/
Dresources.py62 return ops.get_collection(ops.GraphKeys.RESOURCES)
67 return ops.get_collection(ops.GraphKeys.LOCAL_RESOURCES)
Dtemplate.py430 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,
Dvariables.py2814 for var in variable_op.graph.get_collection(collection_name):
3069 return ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES, scope)
3093 return (ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES, scope) +
3094 ops.get_collection(ops.GraphKeys.SAVEABLE_OBJECTS, scope))
3122 return ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES, scope)
3139 return ops.get_collection(ops.GraphKeys.MODEL_VARIABLES, scope)
3161 return ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES, scope)
3183 return ops.get_collection(ops.GraphKeys.MOVING_AVERAGE_VARIABLES, scope)
Dlogging_ops.py606 summary_ops = ops.get_collection(key)
624 summary_op = ops.get_collection(ops.GraphKeys.SUMMARY_OP)
/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)])
Dfeature_column_test.py856 [], ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES))
857 self.assertItemsEqual([], ops.get_collection('my_weights'))
1310 return ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES,
1727 my_vars = g.get_collection('my-vars')
1741 my_vars = g.get_collection('my-vars')
1754 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1765 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1776 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1786 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1804 my_vars = g.get_collection('my-vars')
[all …]
Dfeature_column_v2_test.py68 return ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES,
1995 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
2007 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
2018 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
2029 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
2744 my_vars = g.get_collection('my-vars')
2757 my_vars = g.get_collection('my-vars')
2770 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
2781 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
2792 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
[all …]
/external/tensorflow/tensorflow/python/layers/
Dcore_test.py84 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 2)
113 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 1)
128 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 0)
192 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
203 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
215 len(ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)), 1)
219 len(ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)), 1)
227 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
239 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 2)
Dconvolutional_test.py218 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
230 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
472 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
484 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
496 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
724 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
736 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
748 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
889 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
901 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
[all …]
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_scope_test.py267 ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES),
270 ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES),
273 ops.get_collection(ops.GraphKeys.CONCATENATED_VARIABLES), [concat])
279 self.assertFalse(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES))
280 self.assertFalse(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES))
378 losses = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
394 losses = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
405 losses = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
1130 ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES))
1131 self.assertIn(local_var, ops.get_collection("foo"))
[all …]
Dmetrics_test.py167 set(v.name for v in ops.get_collection(ops.GraphKeys.METRIC_VARIABLES)))
189 self.assertListEqual(ops.get_collection(my_collection_name), [mean])
196 self.assertListEqual(ops.get_collection(my_collection_name), [update_op])
370 self.assertListEqual(ops.get_collection(my_collection_name), [mean])
377 self.assertListEqual(ops.get_collection(my_collection_name), [update_op])
576 self.assertListEqual(ops.get_collection(my_collection_name), [mean])
585 self.assertListEqual(ops.get_collection(my_collection_name), [update_op])
772 self.assertListEqual(ops.get_collection(my_collection_name), [mean])
781 self.assertListEqual(ops.get_collection(my_collection_name), [update_op])
972 self.assertListEqual(ops.get_collection(my_collection_name), [mean])
[all …]
/external/tensorflow/tensorflow/python/framework/
Dmeta_graph_test.py114 new_input_tensor = ops.get_collection("input_tensor")[0]
115 new_output_tensor = ops.get_collection("output_tensor")[0]
274 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)
307 var_list = graph2.get_collection(ops.GraphKeys.METRIC_VARIABLES)
637 graph.get_collection(key)
659 newgraph.get_collection(name=key, scope=import_scope)
664 newgraph.get_collection(name=key, scope=import_scope)
[all …]
Dops_test.py1927 self.assertEqual([12, 34], g.get_collection("key"))
1928 self.assertEqual([], g.get_collection("nothing"))
1929 self.assertEqual([27, blank1, blank2], g.get_collection("blah"))
1930 self.assertEqual([blank1], g.get_collection("blah", "prefix"))
1931 self.assertEqual([blank1], g.get_collection("blah", ".*x"))
1936 other_collection_snapshot = g.get_collection("other")
1943 self.assertEqual(["foo", "bar"], g.get_collection("other"))
1949 empty_coll = g.get_collection("empty")
1959 self.assertEqual(["something"], g.get_collection("empty"))
1967 self.assertEqual(["key"], g.get_collection(1))
[all …]
/external/tensorflow/tensorflow/python/training/
Dtraining_util.py90 global_step_tensors = graph.get_collection(ops.GraphKeys.GLOBAL_STEP)
202 global_step_read_tensors = graph.get_collection(GLOBAL_STEP_READ_KEY)
Dqueue_runner_impl.py468 queue_runners = ops.get_collection(collection)
477 for qr in ops.get_collection(collection):
Dwarm_starting_util.py345 list_of_vars = ops.get_collection(
351 list_of_vars += ops.get_collection(
/external/tensorflow/tensorflow/python/ops/losses/
Dutil.py204 return ops.get_collection(loss_collection, scope)
217 return ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES, scope)
/external/tensorflow/tensorflow/python/compiler/xla/
Djit.py109 xla_scope_counter = ops.get_collection(_XLA_SCOPE_KEY)
/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_v2_test.py163 global_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
337 global_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
546 global_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)

1234