Searched refs:all_vars (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | keras_optimizer_v2_test.py | 61 all_vars = []
76 if not all_vars:
77 all_vars.append(var)
78 all_vars.append(optimizer.get_slot(var, 'm'))
79 all_vars.append(optimizer.get_slot(var, 'v'))
91 self.assertAllClose(1.99, self.evaluate(all_vars[0]))
93 self.assertAllClose(1.2, self.evaluate(all_vars[1]))
95 self.assertAllClose(1.8, self.evaluate(all_vars[2]))
100 self.assertAllClose(1.98, self.evaluate(all_vars[0]))
102 self.assertAllClose(1.44, self.evaluate(all_vars[1]))
[all …]
|
| D | minimize_loss_test.py | 292 all_vars = []
299 all_vars.append(w)
335 v = all_vars[0]
336 self.assertTrue(all(v is vi for vi in all_vars[1:]))
|
| /external/tensorflow/tensorflow/python/training/ |
| D | checkpoint_utils.py | 366 def _collect_partitioned_variable(name, all_vars): argument
368 if name + "/part_0" in all_vars:
371 while name + "/part_%d" % i in all_vars:
372 var.append(all_vars[name + "/part_%d" % i])
|
| D | warm_starting_util_test.py | 81 all_vars = []
84 all_vars.append(variable_scope.get_variable(
89 return [self.evaluate(var) for var in all_vars]
|
| /external/tensorflow/tensorflow/python/layers/ |
| D | normalization_test.py | 89 all_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
90 all_vars_values = [var.eval() for var in all_vars]
702 all_vars = dict([(v.name, v) for v in variables.global_variables()])
703 moving_mean = all_vars['bn/moving_mean:0']
704 moving_variance = all_vars['bn/moving_variance:0']
705 beta = all_vars['bn/beta:0']
706 gamma = all_vars['bn/gamma:0']
765 all_vars = dict([(v.name, v) for v in variables.global_variables()])
766 moving_mean = all_vars['bn/moving_mean:0']
767 moving_variance = all_vars['bn/moving_variance:0']
[all …]
|
| /external/tensorflow/tensorflow/contrib/factorization/python/ops/ |
| D | factorization_ops.py | 321 all_vars = self._row_factors + self._col_factors
322 all_vars.extend([self._row_gramian, self._col_gramian])
325 all_vars.extend(self._row_weights + self._col_weights)
326 return variables.variables_initializer(all_vars)
|
| /external/tensorflow/tensorflow/contrib/rnn/python/kernel_tests/ |
| D | rnn_test.py | 426 all_vars = variables.global_variables()
428 scope_vars = [v for v in all_vars if v.name.startswith(prefix + "/")]
433 self.assertEqual(len(scope_vars), len(all_vars))
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | gradients_test.py | 934 all_vars = vs.global_variables()
935 assert len(all_vars) == 1
936 grads = gradients.gradients(y, [x, all_vars[0]])
1062 all_vars = vs.global_variables()
1063 assert len(all_vars) == 1
1064 grads = gradients.gradients(y, [x, all_vars[0]])
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 334 all_vars = variables_lib.global_variables()
336 scope_vars = [v for v in all_vars if v.name.startswith(prefix + "/")]
341 self.assertEqual(len(scope_vars), len(all_vars))
1585 all_vars = variables_lib.global_variables()
1587 scope_vars = [v for v in all_vars if v.name.startswith(prefix + "/")]
1592 self.assertEqual(len(scope_vars), len(all_vars))
1908 all_vars = variables_lib.global_variables()
1910 scope_vars = [v for v in all_vars if v.name.startswith(prefix + "/")]
1915 self.assertEqual(len(scope_vars), len(all_vars))
2009 all_vars = variables_lib.global_variables()
[all …]
|
| /external/tensorflow/tensorflow/python/grappler/ |
| D | layout_optimizer_test.py | 251 all_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
252 all_vars_values = [var.eval(session=sess) for var in all_vars]
|