Home
last modified time | relevance | path

Searched refs:all_vars (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/python/keras/distribute/
Dkeras_optimizer_v2_test.py60 all_vars = []
75 if not all_vars:
76 all_vars.append(var)
77 all_vars.append(optimizer.get_slot(var, 'm'))
78 all_vars.append(optimizer.get_slot(var, 'v'))
90 self.assertAllClose(1.99, self.evaluate(all_vars[0]))
92 self.assertAllClose(1.2, self.evaluate(all_vars[1]))
94 self.assertAllClose(1.8, self.evaluate(all_vars[2]))
99 self.assertAllClose(1.98, self.evaluate(all_vars[0]))
101 self.assertAllClose(1.44, self.evaluate(all_vars[1]))
[all …]
Dminimize_loss_test.py337 all_vars = []
342 all_vars.append(w)
387 v = all_vars[0]
388 self.assertTrue(all(v is vi for vi in all_vars[1:]))
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils.py481 def _collect_partitioned_variable(name, all_vars): argument
483 if name + "/part_0" in all_vars:
486 while name + "/part_%d" % i in all_vars:
487 var.append(all_vars[name + "/part_%d" % i])
Dwarm_starting_util_test.py83 all_vars = []
86 all_vars.append(variable_scope.get_variable(
91 return [self.evaluate(var) for var in all_vars]
/external/tensorflow/tensorflow/python/keras/legacy_tf_layers/
Dnormalization_test.py90 all_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
91 all_vars_values = [var.eval() for var in all_vars]
703 all_vars = dict([(v.name, v) for v in variables.global_variables()])
704 moving_mean = all_vars['bn/moving_mean:0']
705 moving_variance = all_vars['bn/moving_variance:0']
706 beta = all_vars['bn/beta:0']
707 gamma = all_vars['bn/gamma:0']
766 all_vars = dict([(v.name, v) for v in variables.global_variables()])
767 moving_mean = all_vars['bn/moving_mean:0']
768 moving_variance = all_vars['bn/moving_variance:0']
[all …]
/external/tensorflow/tensorflow/python/ops/
Dgradients_test.py1165 all_vars = vs.global_variables()
1166 assert len(all_vars) == 1
1167 grads = gradients.gradients(y, [x, all_vars[0]])
1224 all_vars = vs.global_variables()
1225 assert len(all_vars) == 4
1226 grads = gradients.gradients(y, [x] + all_vars)
1379 all_vars = vs.global_variables()
1380 assert len(all_vars) == 1
1381 grads = gradients.gradients(y, [x, all_vars[0]])
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_cell_test.py337 all_vars = variables_lib.global_variables()
339 scope_vars = [v for v in all_vars if v.name.startswith(prefix + "/")]
344 self.assertEqual(len(scope_vars), len(all_vars))
1595 all_vars = variables_lib.global_variables()
1597 scope_vars = [v for v in all_vars if v.name.startswith(prefix + "/")]
1602 self.assertEqual(len(scope_vars), len(all_vars))
1918 all_vars = variables_lib.global_variables()
1920 scope_vars = [v for v in all_vars if v.name.startswith(prefix + "/")]
1925 self.assertEqual(len(scope_vars), len(all_vars))
2019 all_vars = variables_lib.global_variables()
[all …]
/external/pigweed/pw_rpc/py/pw_rpc/console_tools/
Dconsole.py58 all_vars = '\n'.join(sorted(self._variables_without_methods()))
/external/tensorflow/tensorflow/python/grappler/
Dlayout_optimizer_test.py272 all_vars = ops.get_collection(ops.GraphKeys.GLOBAL_VARIABLES)
273 all_vars_values = [var.eval(session=sess) for var in all_vars]