Searched refs:get_collection (Results 1 – 25 of 88) sorted by relevance
1234
/external/tensorflow/tensorflow/python/eager/ |
D | function_defun_collection_test.py | 51 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)
|
D | wrap_function_test.py | 215 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/ |
D | plugin_asset.py | 68 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/ |
D | dense_features_test.py | 448 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 …]
|
D | dense_features_v2_test.py | 428 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/ |
D | saved_model_test.py | 82 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 …]
|
D | simple_save.py | 88 assets_collection=ops.get_collection(ops.GraphKeys.ASSET_FILEPATHS),
|
/external/tensorflow/tensorflow/python/ops/ |
D | resources.py | 62 return ops.get_collection(ops.GraphKeys.RESOURCES) 67 return ops.get_collection(ops.GraphKeys.LOCAL_RESOURCES)
|
D | template.py | 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,
|
D | variables.py | 2879 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)
|
D | logging_ops.py | 611 summary_ops = ops.get_collection(key) 629 summary_op = ops.get_collection(ops.GraphKeys.SUMMARY_OP)
|
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/ |
D | core_test.py | 85 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)
|
D | convolutional_test.py | 218 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/ |
D | variable_scope_test.py | 267 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 …]
|
D | metrics_test.py | 167 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/ |
D | meta_graph_test.py | 114 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 …]
|
D | ops_test.py | 2112 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/ |
D | feature_column_test.py | 839 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/ |
D | training_util.py | 90 global_step_tensors = graph.get_collection(ops.GraphKeys.GLOBAL_STEP) 202 global_step_read_tensors = graph.get_collection(GLOBAL_STEP_READ_KEY)
|
D | warm_starting_util.py | 345 list_of_vars = ops.get_collection( 351 list_of_vars += ops.get_collection(
|
D | queue_runner_impl.py | 468 queue_runners = ops.get_collection(collection) 477 for qr in ops.get_collection(collection):
|
/external/tensorflow/tensorflow/python/ops/losses/ |
D | util.py | 203 return ops.get_collection(loss_collection, scope) 216 return ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES, scope)
|
/external/tensorflow/tensorflow/python/compiler/xla/ |
D | jit.py | 137 xla_scope_counter = ops.get_collection(_XLA_SCOPE_KEY)
|
/external/tensorflow/tensorflow/python/tpu/ |
D | feature_column_v2_test.py | 200 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/ |
D | hash_table_asset_v1.py | 75 }, tf.tables_initializer(), tf.get_collection(tf.GraphKeys.ASSET_FILEPATHS)
|
1234