HomeSort by relevance Sort by last modified time
    Searched refs:get_collection (Results 1 - 25 of 122) sorted by null

1 2 3 4 5

  /external/tensorflow/tensorflow/python/summary/
plugin_asset.py 66 container = graph.get_collection(name)
101 for name in graph.get_collection(_PLUGIN_ASSET_PREFIX):
102 collection = graph.get_collection(_PLUGIN_ASSET_PREFIX + name)
  /external/tensorflow/tensorflow/contrib/model_pruning/python/layers/
layers_test.py 50 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/python/ops/losses/
util.py 59 return ops.get_collection(loss_collection, scope)
72 return ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES, scope)
  /external/tensorflow/tensorflow/contrib/gan/python/eval/python/
summaries_test.py 81 len(ops.get_collection(ops.GraphKeys.SUMMARIES)))
100 len(ops.get_collection(ops.GraphKeys.SUMMARIES)))
116 len(ops.get_collection(ops.GraphKeys.SUMMARIES)))
132 len(ops.get_collection(ops.GraphKeys.SUMMARIES)))
  /external/tensorflow/tensorflow/python/ops/
resources.py 61 return ops.get_collection(ops.GraphKeys.RESOURCES)
66 return ops.get_collection(ops.GraphKeys.LOCAL_RESOURCES)
template.py 299 vars_at_start = len(ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES))
301 ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES))
308 trainable_variables = ops.get_collection(
321 variables = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
396 return ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES,
413 return ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES,
422 return ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES,
  /external/tensorflow/tensorflow/python/saved_model/
saved_model_test.py 77 asset_collection = ops.get_collection(ops.GraphKeys.ASSET_FILEPATHS)
181 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
246 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
253 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
260 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
267 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
274 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval())
320 collection_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
330 collection_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
393 42, ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)[0].eval()
    [all...]
simple_save.py 83 assets_collection=ops.get_collection(ops.GraphKeys.ASSET_FILEPATHS),
  /external/tensorflow/tensorflow/contrib/gan/python/losses/python/
losses_impl_test.py 66 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/contrib/layers/python/layers/
summaries_test.py 50 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)]
layers_test.py 82 output_collected = ops.get_collection('outputs')[0]
153 output_collected = ops.get_collection('outputs')[0]
219 output_collected = ops.get_collection('outputs')[0]
439 output_collected = ops.get_collection('outputs')[0]
466 wd = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)[0]
477 ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES), [])
507 len(ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)), 1)
511 len(ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)), 1)
    [all...]
summaries.py 145 for op in ops.get_collection(collection):
  /external/tensorflow/tensorflow/contrib/compiler/
jit.py 95 xla_scope_counter = ops.get_collection(_XLA_SCOPE_KEY)
  /external/tensorflow/tensorflow/contrib/quantize/python/
quantize_graph_test.py 78 [v.name for v in graph.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)])
82 q_variables = graph.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
95 [v.name for v in g.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)])
98 q_variables = g.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
quant_ops_test.py 77 min_max_vars = ops.get_collection(_MIN_MAX_VARS)
  /external/tensorflow/tensorflow/contrib/slim/python/slim/
queues.py 61 for qr in ops.get_collection(ops.GraphKeys.QUEUE_RUNNERS):
summaries_test.py 68 self.assertTrue(op in ops.get_collection(ops.GraphKeys.SUMMARIES))
75 self.assertTrue(op in ops.get_collection(ops.GraphKeys.SUMMARIES))
  /external/tensorflow/tensorflow/contrib/session_bundle/
exporter_test.py 93 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/python/layers/
core_test.py 82 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 2)
103 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 1)
118 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 0)
180 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
190 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
200 len(ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)), 1)
204 len(ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)), 1)
211 loss_keys = ops.get_collection(ops.GraphKeys.REGULARIZATION_LOSSES)
221 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 2)
  /external/tensorflow/tensorflow/python/framework/
meta_graph_test.py 113 new_input_tensor = ops.get_collection("input_tensor")[0]
114 new_output_tensor = ops.get_collection("output_tensor")[0]
268 global_vars = graph1.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
269 trainable_vars = graph1.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
280 global_vars = graph2.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
281 trainable_vars = graph2.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)
564 graph.get_collection(key)
586 newgraph.get_collection(name=key, scope=import_scope)
591 newgraph.get_collection(name=key, scope=import_scope)
    [all...]
  /external/tensorflow/tensorflow/contrib/bayesflow/python/ops/
variable_utils_impl.py 54 tf.get_collection(tf.GraphKeys.TRAINABLE_RESOURCE_VARIABLES)`.
138 ops.get_collection(ops.GraphKeys.TRAINABLE_RESOURCE_VARIABLES))
  /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
cudnn_rnn_ops_benchmark.py 136 trainable_variables = ops.get_collection(
160 trainable_variables = ops.get_collection(
  /external/tensorflow/tensorflow/contrib/eager/python/
metrics_test.py 52 set(ops.get_collection(ops.GraphKeys.LOCAL_VARIABLES)))
55 set(ops.get_collection(ops.GraphKeys.METRIC_VARIABLES)))
  /external/tensorflow/tensorflow/contrib/framework/python/ops/
critical_section_test.py 105 cs, ops.get_collection(critical_section_ops.CRITICAL_SECTIONS))
110 ops.get_collection(critical_section_ops.CRITICAL_SECTION_EXECUTIONS)])
161 # restored_cs = ops.get_collection(critical_section_ops.CRITICAL_SECTIONS)
162 # restored_exec = ops.get_collection(
  /external/tensorflow/tensorflow/contrib/learn/python/learn/utils/
export.py 40 elements = ops.get_collection(collection_name)
80 assets_collection=ops.get_collection(ops.GraphKeys.ASSET_FILEPATHS))

Completed in 613 milliseconds

1 2 3 4 5