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

1 2 3

  /external/tensorflow/tensorflow/python/ops/
template.py 301 ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES))
308 trainable_variables = ops.get_collection(
309 ops.GraphKeys.TRAINABLE_VARIABLES)
312 if trainable_at_start != len(trainable_variables):
316 (trainable_variables[trainable_at_start:],))
393 def trainable_variables(self): member in class:Template
396 return ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES,
406 trainable_variables = set(self.trainable_variables)
407 return [x for x in global_variables if x not in trainable_variables]
497 def trainable_variables(self): member in class:_EagerTemplateVariableStore
670 def trainable_variables(self): member in class:EagerTemplate
    [all...]
  /external/tensorflow/tensorflow/contrib/layers/python/layers/
rev_block_lib_test.py 128 fg_vars = vs.trainable_variables()
223 [x] + variables.trainable_variables())
262 recompute_vars = vs.trainable_variables()
266 reg_vars = vs.trainable_variables()
293 def grad_fn(inputs, trainable_variables, outputs, grad_outputs):
299 outputs, trainable_variables, grad_ys=grad_outputs)
317 loss, [a, b, c] + [variables.trainable_variables()[0]])
319 custom_loss, [a, b, c] + [variables.trainable_variables()[1]])
334 def grad_fn(inputs, trainable_variables, unused_outputs,
341 for i, t in enumerate(trainable_variables)
    [all...]
  /external/tensorflow/tensorflow/compiler/tests/
momentum_test.py 58 self.assertFalse(slot0 in variables.trainable_variables())
61 self.assertFalse(slot1 in variables.trainable_variables())
138 self.assertFalse(slot0 in variables.trainable_variables())
141 self.assertFalse(slot1 in variables.trainable_variables())
  /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/
cudnn_rnn_ops_benchmark.py 136 trainable_variables = ops.get_collection(
137 ops.GraphKeys.TRAINABLE_VARIABLES)
139 trainable_variables)
160 trainable_variables = ops.get_collection(
161 ops.GraphKeys.TRAINABLE_VARIABLES)
163 trainable_variables)
cudnn_rnn_test.py 310 self.assertEqual(1, len(variables.trainable_variables()))
313 variables.trainable_variables()[0].op.name)
538 opaque_params = rnn.trainable_variables[0]
582 opaque_params = (model1.rnn.trainable_variables[0],
583 model2.rnn.trainable_variables[0])
637 assert len(rnn.trainable_variables) == 1
639 rnn.trainable_variables[0],
641 array_ops.shape(rnn.trainable_variables[0]), dtype=dtype))
    [all...]
  /external/tensorflow/tensorflow/python/training/
adadelta_test.py 80 self.assertFalse(slot[0] in variables.trainable_variables())
84 self.assertFalse(slot_update[0] in variables.trainable_variables())
88 self.assertFalse(slot[1] in variables.trainable_variables())
92 self.assertFalse(slot_update[1] in variables.trainable_variables())
moving_averages.py 377 var_list = variables.trainable_variables()
487 variables.moving_average_variables() + variables.trainable_variables()
498 moving_avg_variables = variables.trainable_variables()
moving_averages_test.py 143 self.assertFalse(avg0 in variables.trainable_variables())
144 self.assertFalse(avg1 in variables.trainable_variables())
145 self.assertFalse(avg2 in variables.trainable_variables())
  /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/
fused_rnn_cell_test.py 56 v for v in variables.trainable_variables()
71 v for v in variables.trainable_variables()
92 v for v in variables.trainable_variables()
127 v for v in variables.trainable_variables()
148 v for v in variables.trainable_variables()
lstm_ops_test.py 186 for v in variables.trainable_variables()
197 for v in variables.trainable_variables()
208 for v in variables.trainable_variables()
393 v for v in variables.trainable_variables()
  /external/tensorflow/tensorflow/python/layers/
convolutional_test.py 254 self.assertEqual(len(variables.trainable_variables()), 2)
256 self.assertEqual(len(variables.trainable_variables()), 2)
263 self.assertEqual(len(variables.trainable_variables()), 2)
266 self.assertEqual(len(variables.trainable_variables()), 2)
275 weights = variables.trainable_variables()
290 self.assertEqual(len(variables.trainable_variables()), 2)
292 self.assertEqual(len(variables.trainable_variables()), 4)
415 self.assertEqual(len(variables.trainable_variables()), 3)
417 self.assertEqual(len(variables.trainable_variables()), 3)
424 self.assertEqual(len(variables.trainable_variables()), 3
    [all...]
core_test.py 77 self.assertListEqual(dense.trainable_variables,
82 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 2)
99 self.assertListEqual(dense.trainable_variables, [dense.kernel])
103 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 1)
115 self.assertListEqual(dense.trainable_variables, [])
118 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 0)
221 len(ops.get_collection(ops.GraphKeys.TRAINABLE_VARIABLES)), 2)
238 vars1 = variables.trainable_variables()
240 vars2 = variables.trainable_variables()
249 vars1 = variables.trainable_variables()
    [all...]
  /external/tensorflow/tensorflow/contrib/bayesflow/python/kernel_tests/
custom_grad_test.py 87 gz = gradients_impl.gradients(fz, variables.trainable_variables())
114 gx = gradients_impl.gradients(fx, variables.trainable_variables())
148 gx = gradients_impl.gradients(fx, variables.trainable_variables())
  /external/tensorflow/tensorflow/python/kernel_tests/
functional_ops_test.py 78 self.assertEqual(len(variables.trainable_variables()), 1)
79 self.assertEqual(variables.trainable_variables()[0].name,
87 self.assertEqual(len(variables.trainable_variables()), 1)
113 self.assertEqual(len(variables.trainable_variables()), 1)
114 self.assertEqual(variables.trainable_variables()[0].name,
122 self.assertEqual(len(variables.trainable_variables()), 1)
180 self.assertEqual(len(variables.trainable_variables()), 1)
181 self.assertEqual(variables.trainable_variables()[0].name,
189 self.assertEqual(len(variables.trainable_variables()), 1)
329 self.assertEqual(len(variables.trainable_variables()), 1
    [all...]
template_test.py 329 self.assertEqual(nested1.trainable_variables, [v1])
330 self.assertEqual(nested2.trainable_variables, [v2])
346 self.assertEqual(tmpl1.trainable_variables, [v1, v2])
352 self.assertSequenceEqual(tmpl2.trainable_variables, [v5, v6])
391 self.assertEqual(nested1.trainable_variables, v1)
392 self.assertEqual(nested2.trainable_variables, v2)
609 # Make sure trainable_variables are created.
616 self.assertEqual([], list(ta.trainable_variables))
617 self.assertEqual([], list(tb.trainable_variables))
622 self.assertEqual(1, len(ta.trainable_variables))
    [all...]
rnn_test.py 245 trainable_variables = ops_lib.get_collection(
246 ops_lib.GraphKeys.TRAINABLE_VARIABLES)
248 trainable_variables)
264 trainable_variables = ops_lib.get_collection(
265 ops_lib.GraphKeys.TRAINABLE_VARIABLES)
267 trainable_variables)
378 trainable_variables = ops_lib.get_collection(
379 ops_lib.GraphKeys.TRAINABLE_VARIABLES)
381 trainable_variables)
446 trainable_variables = ops_lib.get_collection
    [all...]
variables_test.py 290 ops.GraphKeys.TRAINABLE_VARIABLES, ops.GraphKeys.GLOBAL_VARIABLES
294 self.assertEqual([var_x, var_z, var_t], variables.trainable_variables())
307 self.assertEqual([var_x, var_y], variables.trainable_variables())
308 self.assertEqual([var_x], variables.trainable_variables("scope_1"))
309 self.assertEqual([var_y], variables.trainable_variables("scope_2"))
  /external/tensorflow/tensorflow/contrib/gan/python/eval/python/
classifier_metrics_test.py 177 self.assertListEqual([], variables.trainable_variables())
192 self.assertListEqual([], variables.trainable_variables())
212 self.assertListEqual([], variables.trainable_variables())
224 self.assertListEqual([], variables.trainable_variables())
236 self.assertListEqual([], variables.trainable_variables())
  /external/tensorflow/tensorflow/contrib/opt/python/training/
elastic_average_optimizer.py 188 under the key `GraphKey.TRAINABLE_VARIABLES`.
206 var_list = variables.trainable_variables()
210 for v, lv in zip(variables.trainable_variables(),
315 local_vars = variables.trainable_variables()
  /external/tensorflow/tensorflow/contrib/bayesflow/python/ops/
variable_utils_impl.py 53 By default `possible_ancestor_vars` is `tf.trainable_variables() +
137 variables_ops.trainable_variables() +
  /external/tensorflow/tensorflow/contrib/tensor_forest/hybrid/python/
hybrid_model.py 125 variables.trainable_variables())
  /external/tensorflow/tensorflow/contrib/model_pruning/examples/cifar10/
cifar10_pruning.py 362 for var in tf.trainable_variables():
373 variables_averages_op = variable_averages.apply(tf.trainable_variables())
  /external/tensorflow/tensorflow/contrib/eager/python/
network_test.py 102 self.assertEqual(1, len(net.trainable_variables))
103 self.evaluate(net.trainable_variables[0].assign([[17.0]]))
113 self.evaluate(net.trainable_variables[0].assign([[17.0]]))
131 self.evaluate(net.trainable_variables[0].assign([[17.0]]))
153 self.evaluate(net1.trainable_variables[0].assign([[17.0]]))
    [all...]
  /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/layers/
cudnn_rnn.py 494 opaque_params=self.trainable_variables[0],
501 name="%s_saveable" % self.trainable_variables[0].name.split(":")[0])
  /external/tensorflow/tensorflow/contrib/training/python/training/
training.py 394 default to all tf.trainable_variables().
431 # Default to tf.trainable_variables()
432 variables_to_train = tf_variables.trainable_variables()
434 # Make sure that variables_to_train are in tf.trainable_variables()
436 assert v in tf_variables.trainable_variables()

Completed in 1159 milliseconds

1 2 3