Home
last modified time | relevance | path

Searched refs:save_counter (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/python/training/
Dcheckpoint_management.py671 save_counter = self._checkpoint.save_counter
672 save_counter.assign_add(1)
684 save_counter = self._checkpoint.save_counter
686 self._save_counter_assign = save_counter.assign_add(1, read_value=False)
689 checkpoint_number = save_counter
Dcheckpoint_management_test.py470 self.assertEqual(2, self.evaluate(checkpoint.save_counter))
550 self.assertEqual(5, self.evaluate(checkpoint.save_counter))
/external/tensorflow/tensorflow/python/training/tracking/
Dutil.py1304 def save_counter(self): member in CheckpointV1
1356 session.run(self.save_counter.initializer)
1358 with ops.colocate_with(self.save_counter):
1359 assign_op = self.save_counter.assign_add(1, read_value=True)
1612 def save_counter(self): member in Checkpoint
1660 session.run(self.save_counter.initializer)
1662 with ops.colocate_with(self.save_counter):
1663 assign_op = self.save_counter.assign_add(1, read_value=True)
Dutil_with_v1_optimizers_test.py202 root_trackable.save_counter # pylint: disable=pointless-statement
212 self.evaluate(state_ops.assign(root_trackable.save_counter, 3))
219 self.assertAllEqual(1, self.evaluate(root_trackable.save_counter))
238 self.assertAllEqual(1, self.evaluate(on_create_root.save_counter))
385 session.run(root.save_counter))
417 self.evaluate(root.save_counter))
460 self.evaluate(root.save_counter))
Dutil_test.py409 root_trackable.save_counter # pylint: disable=pointless-statement
419 self.evaluate(state_ops.assign(root_trackable.save_counter, 3))
427 self.assertAllEqual(1, self.evaluate(root_trackable.save_counter))
442 self.assertAllEqual(1, self.evaluate(on_create_root.save_counter))
529 session.run(root.save_counter))
563 self.evaluate(root.save_counter))
576 self.evaluate(checkpoint.save_counter.assign(12))
600 self.assertEqual(12, self.evaluate(checkpoint.save_counter))
605 self.assertEqual(12, self.evaluate(checkpoint.save_counter))
661 self.evaluate(root.save_counter))
/external/tensorflow/tensorflow/contrib/optimizer_v2/
Dcheckpointable_utils_test.py220 root_trackable.save_counter # pylint: disable=pointless-statement
230 self.evaluate(state_ops.assign(root_trackable.save_counter, 3))
237 self.assertAllEqual(1, self.evaluate(root_trackable.save_counter))
253 self.assertAllEqual(1, self.evaluate(on_create_root.save_counter))
329 session.run(root.save_counter))
363 self.evaluate(root.save_counter))
407 self.evaluate(root.save_counter))
/external/tensorflow/tensorflow/tools/api/golden/v2/
Dtensorflow.train.-checkpoint.pbtxt8 name: "save_counter"
/external/tensorflow/tensorflow/tools/api/golden/v1/
Dtensorflow.train.-checkpoint.pbtxt8 name: "save_counter"
/external/tensorflow/tensorflow/examples/tf2_showcase/
Dmnist.py225 (checkpoint.save_counter.numpy() + 1,
/external/tensorflow/tensorflow/contrib/eager/python/examples/gan/
Dmnist.py306 (checkpoint.save_counter.numpy(),