Searched refs:root_trackable (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/contrib/optimizer_v2/ |
| D | checkpointable_utils_test.py | 104 root_trackable = util.Checkpoint(
120 root_trackable))
123 root_trackable).serialize_object_graph()
211 root_trackable = util.Checkpoint(
220 root_trackable.save_counter # pylint: disable=pointless-statement
222 root_trackable))
228 save_path = root_trackable.save(file_prefix=prefix)
230 self.evaluate(state_ops.assign(root_trackable.save_counter, 3))
234 status = root_trackable.restore(save_path=save_path).assert_consumed()
237 self.assertAllEqual(1, self.evaluate(root_trackable.save_counter))
[all …]
|
| /external/tensorflow/tensorflow/python/training/tracking/ |
| D | util_with_v1_optimizers_test.py | 86 root_trackable = trackable_utils.Checkpoint(
102 root_trackable))
105 root_trackable).serialize_object_graph()
193 root_trackable = trackable_utils.Checkpoint(
202 root_trackable.save_counter # pylint: disable=pointless-statement
204 root_trackable))
210 save_path = root_trackable.save(file_prefix=prefix)
212 self.evaluate(state_ops.assign(root_trackable.save_counter, 3))
216 status = root_trackable.restore(save_path=save_path).assert_consumed()
219 self.assertAllEqual(1, self.evaluate(root_trackable.save_counter))
[all …]
|
| D | util_test.py | 267 root_trackable = trackable_utils.Checkpoint(
285 root_trackable))
288 root_trackable).serialize_object_graph()
401 root_trackable = trackable_utils.Checkpoint(
409 root_trackable.save_counter # pylint: disable=pointless-statement
411 root_trackable))
417 save_path = root_trackable.save(file_prefix=prefix)
419 self.evaluate(state_ops.assign(root_trackable.save_counter, 3))
424 status = root_trackable.restore(save_path=save_path).assert_consumed()
427 self.assertAllEqual(1, self.evaluate(root_trackable.save_counter))
[all …]
|
| D | util.py | 337 def list_objects(root_trackable): argument
351 return graph_view_lib.ObjectGraphView(root_trackable).list_objects()
354 def gather_initializers(root_trackable): argument
368 trackable_objects = list_objects(root_trackable)
1104 def frozen_saver(root_trackable): argument
1126 root_trackable).frozen_saveable_objects()
|
| /external/tensorflow/tensorflow/python/training/ |
| D | saver_test.py | 2926 root_trackable = trackable_utils.Checkpoint(
2932 root_trackable))
2941 return root_trackable
2943 def _set_sentinels(self, root_trackable): argument
2944 self.evaluate(root_trackable.model._named_dense.bias.assign([101.]))
2946 root_trackable.optimizer.get_slot(
2947 var=root_trackable.model._named_dense.bias, name="m")
2949 beta1_power, _ = root_trackable.optimizer._get_beta_accumulators()
2952 def _check_sentinels(self, root_trackable): argument
2954 [1.], self.evaluate(root_trackable.model._named_dense.bias))
[all …]
|