Home
last modified time | relevance | path

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

123456

/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.py123 len(f_wrapped.graph.get_collection(ops.GraphKeys.LOSSES)), 1)
124 f_var_collection = f_wrapped.graph.get_collection(
139 len(g_wrapped.graph.get_collection(ops.GraphKeys.LOSSES)), 1)
140 g_var_collection = g_wrapped.graph.get_collection(
146 self.assertNotEqual(g_wrapped.graph.get_collection(ops.GraphKeys.LOSSES),
147 f_wrapped.graph.get_collection(ops.GraphKeys.LOSSES))
292 self.assertLen(graph.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES), 1)
293 self.assertLen(graph.get_collection('a'), 2)
294 self.assertLen(graph.get_collection('b'), 1)
/external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
Dlayers_test.py50 masks = ops.get_collection(core_layers.MASK_COLLECTION)
55 masked_weight = ops.get_collection(core_layers.MASKED_WEIGHT_COLLECTION)
75 masks = ops.get_collection(core_layers.MASK_COLLECTION)
83 masked_weight = ops.get_collection(core_layers.MASKED_WEIGHT_COLLECTION)
99 masks = ops.get_collection(core_layers.MASK_COLLECTION)
104 masked_weight = ops.get_collection(core_layers.MASKED_WEIGHT_COLLECTION)
123 masks = ops.get_collection(core_layers.MASK_COLLECTION)
130 masked_weight = ops.get_collection(core_layers.MASKED_WEIGHT_COLLECTION)
/external/tensorflow/tensorflow/contrib/gan/python/losses/python/
Dlosses_impl_test.py66 self.assertEqual(0, len(ops.get_collection('collection')))
69 self.assertEqual(1, len(ops.get_collection('collection')))
72 self.assertEqual(0, len(ops.get_collection('collection')))
76 self.assertEqual(1, len(ops.get_collection('collection')))
168 self.assertEqual(0, len(ops.get_collection(ops.GraphKeys.SUMMARIES)))
170 self.assertLess(0, len(ops.get_collection(ops.GraphKeys.SUMMARIES)))
173 self.assertEqual(0, len(ops.get_collection(ops.GraphKeys.SUMMARIES)))
177 self.assertLess(0, len(ops.get_collection(ops.GraphKeys.SUMMARIES)))
299 self.assertEqual(0, len(ops.get_collection('collection')))
301 self.assertEqual(1, len(ops.get_collection('collection')))
[all …]
/external/tensorflow/tensorflow/python/saved_model/
Dsaved_model_test.py78 asset_collection = ops.get_collection(ops.GraphKeys.ASSET_FILEPATHS)
221 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
288 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
295 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
302 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
309 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
316 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
363 collection_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
373 collection_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
438 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
[all …]
/external/tensorflow/tensorflow/contrib/gan/python/eval/python/
Dsummaries_test.py108 len(ops.get_collection(ops.GraphKeys.SUMMARIES)))
128 len(ops.get_collection(ops.GraphKeys.SUMMARIES)))
144 len(ops.get_collection(ops.GraphKeys.SUMMARIES)))
160 len(ops.get_collection(ops.GraphKeys.SUMMARIES)))
170 self.assertEquals(2, len(ops.get_collection(ops.GraphKeys.SUMMARIES)))
178 self.assertEquals(1, len(ops.get_collection(ops.GraphKeys.SUMMARIES)))
/external/tensorflow/tensorflow/contrib/gan/python/features/python/
Dspectral_normalization_test.py207 weight_vars = ops.get_collection('CONTRIB_LAYERS_CONV2D_WEIGHTS')
209 bias_vars = ops.get_collection('CONTRIB_LAYERS_CONV2D_BIASES')
236 weight_vars = ops.get_collection('SLIM_CONV2D_WEIGHTS')
238 bias_vars = ops.get_collection('SLIM_CONV2D_BIASES')
294 weight_vars = ops.get_collection('CONTRIB_LAYERS_FC_WEIGHTS')
296 bias_vars = ops.get_collection('CONTRIB_LAYERS_FC_BIASES')
323 weight_vars = ops.get_collection('SLIM_FC_WEIGHTS')
325 bias_vars = ops.get_collection('SLIM_FC_BIASES')
/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/contrib/session_bundle/
Dexporter_test.py93 assets_collection = ops.get_collection(ops.GraphKeys.ASSET_FILEPATHS)
207 self.assertEqual(10, ops.get_collection("v")[0].eval())
208 self.assertEqual(20, ops.get_collection("v")[1].eval())
209 ops.get_collection(constants.INIT_OP_KEY)[0].run()
210 self.assertEqual(30, ops.get_collection("v")[2].eval())
/external/tensorflow/tensorflow/contrib/layers/python/layers/
Dlayers_test.py82 output_collected = ops.get_collection('outputs')[0]
153 output_collected = ops.get_collection('outputs')[0]
219 output_collected = ops.get_collection('outputs')[0]
450 output_collected = ops.get_collection('outputs')[0]
477 wd = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)[0]
488 ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES), [])
518 len(ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)), 1)
522 len(ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)), 1)
1393 c_output = ops.get_collection('outputs')[0]
1477 c_output = ops.get_collection('outputs')[0]
[all …]
Dsummaries_test.py50 names = [op.op.name for op in ops.get_collection(ops.GraphKeys.SUMMARIES)]
61 names = [op.op.name for op in ops.get_collection(ops.GraphKeys.SUMMARIES)]
73 names = [op.op.name for op in ops.get_collection(ops.GraphKeys.SUMMARIES)]
/external/tensorflow/tensorflow/python/ops/losses/
Dutil.py54 return ops.get_collection(loss_collection, scope)
67 return ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES, scope)
/external/tensorflow/tensorflow/contrib/eager/python/
Dmetrics_test.py58 set(ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES)))
59 self.assertEqual(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES), [])
62 set(ops.get_collection(ops.GraphKeys.METRIC_VARIABLES)))
70 set(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
71 self.assertEqual(ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES), [])
74 set(ops.get_collection(ops.GraphKeys.METRIC_VARIABLES)))
/external/tensorflow/tensorflow/python/ops/
Dresources.py62 return ops.get_collection(ops.GraphKeys.RESOURCES)
67 return ops.get_collection(ops.GraphKeys.LOCAL_RESOURCES)
/external/tensorflow/tensorflow/contrib/framework/python/ops/
Dvariables_test.py257 self.assertTrue(a in ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES))
258 self.assertFalse(a in ops.get_collection(ops.GraphKeys.MODEL_VARIABLES))
393 loss = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)[0]
401 loss = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)[0]
585 self.assertTrue(a in ops.get_collection(ops.GraphKeys.MODEL_VARIABLES))
676 self.assertEquals(a, ops.get_collection('A')[0])
677 self.assertEquals(b, ops.get_collection('B')[0])
683 self.assertEquals(a, ops.get_collection('A')[0])
684 self.assertEquals(b, ops.get_collection('B')[0])
685 self.assertListEqual([a, b], ops.get_collection('C'))
[all …]
/external/tensorflow/tensorflow/python/layers/
Dcore_test.py83 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 2)
112 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 1)
127 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 0)
191 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
202 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
214 len(ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)), 1)
218 len(ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)), 1)
226 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
238 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 2)
/external/tensorflow/tensorflow/python/kernel_tests/
Dvariable_scope_test.py255 ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES),
258 ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES),
261 ops.get_collection(ops.GraphKeys.CONCATENATED_VARIABLES), [concat])
267 self.assertFalse(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES))
268 self.assertFalse(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES))
366 losses = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
382 losses = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
393 losses = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
1118 ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES))
1119 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.py113 new_input_tensor = ops.get_collection("input_tensor")[0]
114 new_output_tensor = ops.get_collection("output_tensor")[0]
273 global_vars = graph1.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
274 trainable_vars = graph1.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
285 global_vars = graph2.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
286 trainable_vars = graph2.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
614 graph.get_collection(key)
636 newgraph.get_collection(name=key, scope=import_scope)
641 newgraph.get_collection(name=key, scope=import_scope)
929 self.assertEqual(len(ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES)),
Dops_test.py1606 self.assertEqual([12, 34], g.get_collection("key"))
1607 self.assertEqual([], g.get_collection("nothing"))
1608 self.assertEqual([27, blank1, blank2], g.get_collection("blah"))
1609 self.assertEqual([blank1], g.get_collection("blah", "prefix"))
1610 self.assertEqual([blank1], g.get_collection("blah", ".*x"))
1615 other_collection_snapshot = g.get_collection("other")
1622 self.assertEqual(["foo", "bar"], g.get_collection("other"))
1628 empty_coll = g.get_collection("empty")
1638 self.assertEqual(["something"], g.get_collection("empty"))
1646 self.assertEqual(["key"], g.get_collection(1))
[all …]
/external/tensorflow/tensorflow/contrib/slim/python/slim/
Dsummaries_test.py68 self.assertTrue(op in ops.get_collection(ops.GraphKeys.SUMMARIES))
75 self.assertTrue(op in ops.get_collection(ops.GraphKeys.SUMMARIES))
/external/tensorflow/tensorflow/python/feature_column/
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 …]
/external/tensorflow/tensorflow/contrib/model_pruning/python/
Dpruning.py116 return ops.get_collection(_MASKED_WEIGHT_COLLECTION)
120 return ops.get_collection(_MASK_COLLECTION)
124 return ops.get_collection(_THRESHOLD_COLLECTION)
128 return ops.get_collection(_WEIGHT_COLLECTION)
/external/tensorflow/tensorflow/contrib/metrics/python/ops/
Dmetric_ops_test.py156 set(v.name for v in ops.get_collection(ops.GraphKeys.METRIC_VARIABLES)))
172 self.assertListEqual(ops.get_collection(my_collection_name), [mean])
178 self.assertListEqual(ops.get_collection(my_collection_name), [update_op])
331 self.assertListEqual(ops.get_collection(my_collection_name), [mean])
337 self.assertListEqual(ops.get_collection(my_collection_name), [update_op])
499 self.assertListEqual(ops.get_collection(my_collection_name), [mean])
507 self.assertListEqual(ops.get_collection(my_collection_name), [update_op])
1048 self.assertListEqual(ops.get_collection(my_collection_name), [mean])
1056 self.assertListEqual(ops.get_collection(my_collection_name), [update_op])
1213 self.assertListEqual(ops.get_collection(my_collection_name), [mean])
[all …]
/external/tensorflow/tensorflow/python/training/
Dtraining_util.py91 global_step_tensors = graph.get_collection(ops.GraphKeys.GLOBAL_STEP)
204 global_step_read_tensors = graph.get_collection(GLOBAL_STEP_READ_KEY)

123456