| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util_test.py | 52 from tensorflow.python.training.tracking import util as trackable_utils unknown
59 self.a_variable = trackable_utils.add_variable(
94 trackable_utils.add_variable(
96 constant_initializer = trackable_utils.add_variable(
99 ones_initializer = trackable_utils.add_variable(
104 bare_initializer = trackable_utils.add_variable(
115 duplicate = trackable_utils.add_variable(
118 trackable_utils.add_variable(obj, name="duplicate", shape=[])
120 self.evaluate(trackable_utils.gather_initializers(obj))
158 trackable_utils.add_variable(NoInit(), "var", shape=[])
[all …]
|
| D | util_with_v1_optimizers_test.py | 48 from tensorflow.python.training.tracking import util as trackable_utils unknown
55 self.a_variable = trackable_utils.add_variable(
86 root_trackable = trackable_utils.Checkpoint(
101 self.evaluate(trackable_utils.gather_initializers(
193 root_trackable = trackable_utils.Checkpoint(
203 self.evaluate(trackable_utils.gather_initializers(
229 on_create_root = trackable_utils.Checkpoint(
269 root = trackable_utils.Checkpoint(
300 root = trackable_utils.Checkpoint(
331 root = trackable_utils.Checkpoint(
[all …]
|
| D | util_xla_test.py | 29 from tensorflow.python.training.tracking import util as trackable_utils unknown
36 self.a_variable = trackable_utils.add_variable(
65 root = trackable_utils.Checkpoint(
|
| /external/tensorflow/tensorflow/python/data/kernel_tests/ |
| D | iterator_checkpoint_test.py | 31 from tensorflow.python.training.tracking import util as trackable_utils unknown
46 checkpoint = trackable_utils.Checkpoint(iterator=iterator)
76 checkpoint = trackable_utils.Checkpoint(
99 checkpoint = trackable_utils.Checkpoint(iterator=iterator)
118 checkpoint = trackable_utils.Checkpoint(iterator=iterator)
|
| /external/tensorflow/tensorflow/contrib/distribute/python/ |
| D | checkpointing_test.py | 34 from tensorflow.python.training.tracking import util as trackable_utils unknown
41 self.a_variable = trackable_utils.add_variable(
79 root = trackable_utils.Checkpoint(
|
| /external/tensorflow/tensorflow/contrib/checkpoint/python/ |
| D | visualize.py | 22 from tensorflow.python.training.tracking import util as trackable_utils unknown
54 object_graph = trackable_utils.object_metadata(save_path)
|
| D | visualize_test.py | 31 from tensorflow.python.training.tracking import util as trackable_utils unknown
60 save_checkpoint = trackable_utils.Checkpoint(
|
| /external/tensorflow/tensorflow/contrib/eager/python/ |
| D | datasets_test.py | 40 from tensorflow.python.training.tracking import util as trackable_utils unknown
241 checkpoint = trackable_utils.Checkpoint(iterator=iterator)
260 checkpoint = trackable_utils.Checkpoint(
282 checkpoint = trackable_utils.Checkpoint(iterator=iterator)
296 checkpoint = trackable_utils.Checkpoint(iterator=iterator)
|
| D | metrics_test.py | 38 from tensorflow.python.training.tracking import util as trackable_utils unknown
317 checkpoint = trackable_utils.Checkpoint(mean=mean)
330 restore_checkpoint = trackable_utils.Checkpoint(mean=restore_mean)
|
| D | network_test.py | 34 from tensorflow.python.training.tracking import util as trackable_utils unknown
68 checkpoint = trackable_utils.Checkpoint(net=MyNetwork())
|
| /external/tensorflow/tensorflow/contrib/eager/python/examples/spinn/ |
| D | spinn_test.py | 40 from tensorflow.python.training.tracking import util as trackable_utils unknown
424 object_graph = trackable_utils.object_metadata(
|
| /external/tensorflow/tensorflow/contrib/cudnn_rnn/python/kernel_tests/ |
| D | cudnn_rnn_test.py | 61 from tensorflow.python.training.tracking import util as trackable_utils unknown
721 cudnn_checkpoint = trackable_utils.Checkpoint(cell=cudnn_layer)
729 restore_layer_checkpoint = trackable_utils.Checkpoint(
762 save_checkpoint = trackable_utils.Checkpoint(cell=save_cell_layer)
822 save_checkpoint = trackable_utils.Checkpoint(cell=save_layer)
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | network.py | 50 from tensorflow.python.training.tracking import util as trackable_utils unknown
211 trackable_utils.saver_with_op_caching(self))
1485 trackable_utils.streaming_restore(status=status, session=session)
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | metrics_test.py | 42 from tensorflow.python.training.tracking import util as trackable_utils unknown
135 checkpoint = trackable_utils.Checkpoint(sum=m)
153 restore_checkpoint = trackable_utils.Checkpoint(sum=restore_sum)
271 checkpoint = trackable_utils.Checkpoint(mean=m)
289 restore_checkpoint = trackable_utils.Checkpoint(mean=restore_mean)
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver_test.py | 78 from tensorflow.python.training.tracking import util as trackable_utils unknown
2838 self.a_variable = trackable_utils.add_variable(
2926 root_trackable = trackable_utils.Checkpoint(
2931 self.evaluate(trackable_utils.gather_initializers(
3015 object_saver = trackable_utils.Checkpoint(root=root)
3021 second_saver = trackable_utils.Checkpoint(v=var)
3049 object_saver = trackable_utils.Checkpoint(root=root)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | rnn_cell_test.py | 52 from tensorflow.python.training.tracking import util as trackable_utils unknown
2807 checkpoint = trackable_utils.Checkpoint(wrapper=wrapper)
|