Home
last modified time | relevance | path

Searched refs:trackable_utils (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/python/training/tracking/
Dutil_test.py48 from tensorflow.python.training.tracking import util as trackable_utils unknown
55 self.a_variable = trackable_utils.add_variable(
65 trackable_utils.add_variable(
67 constant_initializer = trackable_utils.add_variable(
70 ones_initializer = trackable_utils.add_variable(
75 bare_initializer = trackable_utils.add_variable(
86 duplicate = trackable_utils.add_variable(
89 trackable_utils.add_variable(obj, name="duplicate", shape=[])
91 self.evaluate(trackable_utils.gather_initializers(obj))
129 trackable_utils.add_variable(NoInit(), "var", shape=[])
[all …]
Dutil_with_v1_optimizers_test.py36 from tensorflow.python.training.tracking import util as trackable_utils unknown
45 root = trackable_utils.Checkpoint()
46 root.var = trackable_utils.add_variable(
56 self.evaluate(trackable_utils.gather_initializers(
57 trackable_utils.Checkpoint(root=root, optimizer=optimizer)))
66 new_root = trackable_utils.Checkpoint()
73 new_root.var = trackable_utils.add_variable(
110 obj = trackable_utils.Checkpoint()
114 self.evaluate(trackable_utils.gather_initializers(obj))
127 obj = trackable_utils.Checkpoint()
[all …]
/external/tensorflow/tensorflow/python/keras/tests/
Dtracking_util_test.py52 from tensorflow.python.training.tracking import util as trackable_utils unknown
75 self.a_variable = trackable_utils.add_variable(
98 checkpoint = trackable_utils.Checkpoint(model=model)
113 root_trackable = trackable_utils.Checkpoint(
130 self.evaluate(trackable_utils.gather_initializers(
203 root_trackable = trackable_utils.Checkpoint(
212 self.evaluate(trackable_utils.gather_initializers(
235 on_create_root = trackable_utils.Checkpoint(
276 root = trackable_utils.Checkpoint(
302 root = trackable_utils.CheckpointV1(
[all …]
Dtracking_util_with_v1_optimizers_test.py46 from tensorflow.python.training.tracking import util as trackable_utils unknown
53 self.a_variable = trackable_utils.add_variable(
84 root_trackable = trackable_utils.Checkpoint(
99 self.evaluate(trackable_utils.gather_initializers(
188 root_trackable = trackable_utils.Checkpoint(
198 self.evaluate(trackable_utils.gather_initializers(
225 on_create_root = trackable_utils.Checkpoint(
265 root = trackable_utils.Checkpoint(
296 root = trackable_utils.Checkpoint(
328 root = trackable_utils.Checkpoint(
[all …]
Dsaver_test.py36 from tensorflow.python.training.tracking import util as trackable_utils unknown
43 self.a_variable = trackable_utils.add_variable(
69 root_trackable = trackable_utils.Checkpoint(
74 self.evaluate(trackable_utils.gather_initializers(
111 object_saver = trackable_utils.Checkpoint(root=root)
117 second_saver = trackable_utils.Checkpoint(v=var)
145 object_saver = trackable_utils.Checkpoint(root=root)
Dtracking_util_xla_test.py29 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/keras/distribute/
Dcheckpointing_test.py35 from tensorflow.python.training.tracking import util as trackable_utils unknown
73 checkpoint = trackable_utils.Checkpoint(v=v, opt=opt)
82 checkpoint = trackable_utils.Checkpoint(v=v, opt=opt)
92 checkpoint = trackable_utils.Checkpoint(v=v, opt=opt)
123 checkpoint = trackable_utils.Checkpoint(v=v)
132 checkpoint = trackable_utils.Checkpoint(v=v)
/external/tensorflow/tensorflow/python/distribute/
Dcheckpointing_test.py29 from tensorflow.python.training.tracking import util as trackable_utils unknown
46 save_checkpoint = trackable_utils.Checkpoint(v=variables_lib.Variable(
51 restore_checkpoint = trackable_utils.Checkpoint()
Dvars_test.py47 from tensorflow.python.training.tracking import util as trackable_utils unknown
418 ckpt = trackable_utils.Checkpoint(var=v_on_write)
429 ckpt_normal = trackable_utils.Checkpoint(var=v_normal_restore)
436 ckpt = trackable_utils.Checkpoint(var=v_normal_save)
441 ckpt_on_write = trackable_utils.Checkpoint(var=v_on_write)
1104 ckpt = trackable_utils.Checkpoint(var=v_on_read)
1113 ckpt_normal = trackable_utils.Checkpoint(var=v_normal_restore)
1121 ckpt = trackable_utils.Checkpoint(var=v_normal_save)
1126 ckpt_on_read = trackable_utils.Checkpoint(var=v_on_read)
Dvalues_test.py63 from tensorflow.python.training.tracking import util as trackable_utils unknown
445 checkpoint = trackable_utils.Checkpoint(v=v)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dcheckpoint_test.py44 from tensorflow.python.training.tracking import util as trackable_utils unknown
307 checkpoint = trackable_utils.Checkpoint(iterator=iterator)
332 checkpoint = trackable_utils.Checkpoint(
354 checkpoint = trackable_utils.Checkpoint(iterator=iterator)
373 checkpoint = trackable_utils.Checkpoint(iterator=iterator)
387 ckpt = trackable_utils.Checkpoint(
405 ckpt = trackable_utils.Checkpoint(
555 checkpoint = trackable_utils.Checkpoint(iterator=iterator)
Dshuffle_test.py39 from tensorflow.python.training.tracking import util as trackable_utils unknown
364 ckpt = trackable_utils.Checkpoint(iterator=iterator)
Dcache_test.py40 from tensorflow.python.training.tracking import util as trackable_utils unknown
399 ckpt = trackable_utils.Checkpoint(iterator=iterator)
Dmap_test.py60 from tensorflow.python.training.tracking import util as trackable_utils unknown
1397 ckpt = trackable_utils.Checkpoint(iterator=iterator)
/external/tensorflow/tensorflow/python/training/experimental/
Dloss_scale_optimizer_test.py40 from tensorflow.python.training.tracking import util as trackable_utils unknown
295 checkpoint = trackable_utils.Checkpoint(optimizer=opt)
/external/tensorflow/tensorflow/python/training/
Dcheckpoint_utils_test.py41 from tensorflow.python.training.tracking import util as trackable_utils unknown
421 checkpoint = trackable_utils.Checkpoint(a=a)
/external/tensorflow/tensorflow/python/keras/optimizer_v2/
Doptimizer_v2_test.py62 from tensorflow.python.training.tracking import util as trackable_utils unknown
688 checkpoint = trackable_utils.Checkpoint(optimizer=opt)
694 optimizer_checkpoint = trackable_utils.Checkpoint(iter=iterations_var)
695 checkpoint_to_restore = trackable_utils.Checkpoint(
/external/tensorflow/tensorflow/python/keras/mixed_precision/
Dloss_scale_optimizer_test.py44 from tensorflow.python.training.tracking import util as trackable_utils unknown
677 checkpoint = trackable_utils.Checkpoint(optimizer=opt, var=var)
689 checkpoint = trackable_utils.Checkpoint(optimizer=opt, var=var)
Dautocast_variable_test.py53 from tensorflow.python.training.tracking import util as trackable_utils unknown
458 checkpoint = trackable_utils.Checkpoint(x=x)
Dkeras_test.py60 from tensorflow.python.training.tracking import util as trackable_utils unknown
300 checkpoint = trackable_utils.Checkpoint(layer=layer)
312 checkpoint = trackable_utils.Checkpoint(layer=layer)
/external/tensorflow/tensorflow/python/keras/engine/
Dtraining.py82 from tensorflow.python.training.tracking import util as trackable_utils unknown
321 trackable_utils.saver_with_op_caching(self))
2272 trackable_utils.streaming_restore(status=status, session=session)
/external/tensorflow/tensorflow/python/keras/
Dmetrics_test.py48 from tensorflow.python.training.tracking import util as trackable_utils unknown
143 checkpoint = trackable_utils.Checkpoint(sum=m)
161 restore_checkpoint = trackable_utils.Checkpoint(sum=restore_sum)
299 checkpoint = trackable_utils.Checkpoint(mean=m)
317 restore_checkpoint = trackable_utils.Checkpoint(mean=restore_mean)
/external/tensorflow/tensorflow/python/kernel_tests/
Drnn_cell_test.py55 from tensorflow.python.training.tracking import util as trackable_utils unknown
2819 checkpoint = trackable_utils.Checkpoint(wrapper=wrapper)