Home
last modified time | relevance | path

Searched refs:get_collection (Results 1 – 25 of 88) 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.assertEqual(ops.get_collection('x'), [2])
59 self.assertEqual(ops.get_collection('y'), [5])
60 self.assertEqual(ops.get_collection('z'), [])
79 self.assertLen(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES), 1)
94 self.assertLen(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/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/keras/feature_column/
Ddense_features_test.py448 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
455 [v.name for v in ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)])
486 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
489 [v.name for v in ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)])
519 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
538 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
541 [v.name for v in ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)])
787 global_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
799 trainable_vars = ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
878 global_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
[all …]
Ddense_features_v2_test.py428 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
435 [v.name for v in ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)])
466 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
469 [v.name for v in ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)])
499 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
518 len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)))
521 [v.name for v in ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)])
/external/tensorflow/tensorflow/python/saved_model/
Dsaved_model_test.py82 asset_collection = ops.get_collection(ops.GraphKeys.ASSET_FILEPATHS)
243 self._eval(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0]))
313 self._eval(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0]))
321 self._eval(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0]))
330 self._eval(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0]))
339 self._eval(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0]))
347 self._eval(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0]))
394 collection_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
404 collection_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
500 self._eval(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0]))
[all …]
Dsimple_save.py88 assets_collection=ops.get_collection(ops.GraphKeys.ASSET_FILEPATHS),
/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.py2879 for var in variable_op.graph.get_collection(collection_name):
3133 return ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES, scope)
3157 return (ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES, scope) +
3158 ops.get_collection(ops.GraphKeys.SAVEABLE_OBJECTS, scope))
3186 return ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES, scope)
3203 return ops.get_collection(ops.GraphKeys.MODEL_VARIABLES, scope)
3225 return ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES, scope)
3247 return ops.get_collection(ops.GraphKeys.MOVING_AVERAGE_VARIABLES, scope)
Dlogging_ops.py611 summary_ops = ops.get_collection(key)
629 summary_op = ops.get_collection(ops.GraphKeys.SUMMARY_OP)
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dcore_test.py85 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 2)
114 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 1)
129 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 0)
193 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
204 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
216 len(ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)), 1)
220 len(ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)), 1)
228 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
240 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 2)
Dconvolutional_test.py218 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
231 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
473 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
486 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
499 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
727 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
740 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
753 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
895 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
908 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))
393 losses = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
409 losses = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
420 losses = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
1145 ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES))
1146 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]
297 global_vars = graph1.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
298 trainable_vars = graph1.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
309 global_vars = graph2.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
310 trainable_vars = graph2.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
330 var_list = graph2.get_collection(ops.GraphKeys.METRIC_VARIABLES)
660 graph.get_collection(key)
682 newgraph.get_collection(name=key, scope=import_scope)
687 newgraph.get_collection(name=key, scope=import_scope)
[all …]
Dops_test.py2112 self.assertEqual([12, 34], g.get_collection("key"))
2113 self.assertEqual([], g.get_collection("nothing"))
2114 self.assertEqual([27, blank1, blank2], g.get_collection("blah"))
2115 self.assertEqual([blank1], g.get_collection("blah", "prefix"))
2116 self.assertEqual([blank1], g.get_collection("blah", ".*x"))
2121 other_collection_snapshot = g.get_collection("other")
2128 self.assertEqual(["foo", "bar"], g.get_collection("other"))
2134 empty_coll = g.get_collection("empty")
2144 self.assertEqual(["something"], g.get_collection("empty"))
2152 self.assertEqual(["key"], g.get_collection(1))
[all …]
/external/tensorflow/tensorflow/python/feature_column/
Dfeature_column_test.py839 ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES))
840 self.assertCountEqual([], ops.get_collection('my_weights'))
1282 return ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES,
1699 my_vars = g.get_collection('my-vars')
1713 my_vars = g.get_collection('my-vars')
1726 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1737 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1748 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1758 trainable_vars = g.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
1776 my_vars = g.get_collection('my-vars')
[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)
Dwarm_starting_util.py345 list_of_vars = ops.get_collection(
351 list_of_vars += ops.get_collection(
Dqueue_runner_impl.py468 queue_runners = ops.get_collection(collection)
477 for qr in ops.get_collection(collection):
/external/tensorflow/tensorflow/python/ops/losses/
Dutil.py203 return ops.get_collection(loss_collection, scope)
216 return ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES, scope)
/external/tensorflow/tensorflow/python/compiler/xla/
Djit.py137 xla_scope_counter = ops.get_collection(_XLA_SCOPE_KEY)
/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_v2_test.py200 global_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
424 global_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
634 global_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
706 global_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/tests/tf_saved_model/
Dhash_table_asset_v1.py75 }, tf.tables_initializer(), tf.get_collection(tf.GraphKeys.ASSET_FILEPATHS)

1234