Searched refs:watched_variables (Results 1 – 15 of 15) sorted by relevance
| /external/tensorflow/tensorflow/python/eager/ |
| D | tape_test.py | 180 self.assertAllEqual(variable_watcher.watched_variables(), (var1, var2))
189 self.assertAllEqual(variable_watcher.watched_variables(), (var1,))
201 self.assertAllEqual(variable_watcher1.watched_variables(), (var1, var2))
202 self.assertAllEqual(variable_watcher2.watched_variables(), (var2,))
211 self.assertAllEqual(variable_watcher.watched_variables(), (var1, var2))
|
| D | tape.py | 43 def watched_variables(self): member in Tape
89 def watched_variables(self): member in VariableWatcher
|
| D | backprop.py | 241 variables = this_tape.watched_variables()
981 def watched_variables(self): member in GradientTape
984 self._watched_variables = self._tape.watched_variables()
1084 self._watched_variables = self._tape.watched_variables()
|
| D | backprop_test.py | 463 self.assertEqual(0, len(t.watched_variables()))
466 self.assertAllEqual([v], t.watched_variables())
469 self.assertEqual(0, len(t.watched_variables()))
471 self.assertAllEqual([v], t.watched_variables())
475 self.assertEqual(0, len(t.watched_variables()))
478 self.assertAllEqual([v], t.watched_variables())
481 self.assertEqual(0, len(t.watched_variables()))
483 self.assertAllEqual([v], t.watched_variables())
1380 self.assertTupleEqual(tape.watched_variables(), (x2,))
1570 self.assertEqual((x,), tape.watched_variables())
[all …]
|
| D | function_gradients_test.py | 160 g, = tape.gradient(primal_out, tape.watched_variables())
207 x, = tape.gradient(x, tape.watched_variables())
|
| D | function_test.py | 4986 self.assertEqual((v,), tape.watched_variables())
4995 self.assertEqual((v,), tape.watched_variables())
5008 self.assertEqual((v,), tape.watched_variables())
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-gradient-tape.pbtxt | 34 name: "watched_variables"
|
| D | tensorflow.autodiff.-gradient-tape.pbtxt | 34 name: "watched_variables"
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-gradient-tape.pbtxt | 34 name: "watched_variables"
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | custom_gradient.py | 352 v.ref() for v in variable_watcher.watched_variables()
445 for v in set(v.ref() for v in variable_watcher.watched_variables())
|
| /external/tensorflow/tensorflow/python/keras/layers/ |
| D | core.py | 919 self._check_variables(created_variables, tape.watched_variables())
1361 self._check_variables(created_variables, tape.watched_variables())
|
| /external/tensorflow/tensorflow/python/training/ |
| D | optimizer.py | 479 var_list = tape.watched_variables()
|
| /external/tensorflow/tensorflow/python/saved_model/ |
| D | load_v1_in_v2_test.py | 115 self.assertEqual(imported.variables[:1], list(tape.watched_variables()))
|
| D | load_test.py | 763 self.assertAllEqual(t.watched_variables(), [obj.weight])
791 self.assertAllEqual(t.watched_variables(), [obj.weight])
|
| /external/tensorflow/tensorflow/python/distribute/ |
| D | parameter_server_strategy_test.py | 883 v, = tape.watched_variables()
|