/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()
|