Searched refs:save_counter (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/python/training/ |
D | checkpoint_management.py | 792 save_counter = self._checkpoint.save_counter 793 save_counter.assign_add(1) 805 save_counter = self._checkpoint.save_counter 807 self._save_counter_assign = save_counter.assign_add(1, read_value=False) 810 checkpoint_number = save_counter
|
D | checkpoint_management_test.py | 470 self.assertEqual(2, self.evaluate(checkpoint.save_counter)) 550 self.assertEqual(5, self.evaluate(checkpoint.save_counter))
|
/external/tensorflow/tensorflow/python/training/tracking/ |
D | util.py | 1601 def save_counter(self): member in CheckpointV1 1653 session.run(self.save_counter.initializer) 1655 with ops.colocate_with(self.save_counter): 1656 assign_op = self.save_counter.assign_add(1, read_value=True) 1780 status.add_to_optionally_restored(self.save_counter) 2023 def save_counter(self): member in Checkpoint 2090 session.run(self.save_counter.initializer) 2092 with ops.colocate_with(self.save_counter): 2093 assign_op = self.save_counter.assign_add(1, read_value=True) 2272 status.add_to_optionally_restored(self.save_counter)
|
D | util_test.py | 294 self.evaluate(checkpoint.save_counter.assign(12)) 318 self.assertEqual(12, self.evaluate(checkpoint.save_counter)) 323 self.assertEqual(12, self.evaluate(checkpoint.save_counter)) 850 self.assertEqual(self.evaluate(load_checkpoint.save_counter), 1) 861 self.assertEqual(self.evaluate(load_checkpoint.save_counter), 1) 871 self.assertEqual(self.evaluate(load_checkpoint.save_counter), 1)
|
/external/tensorflow/tensorflow/python/keras/tests/ |
D | tracking_util_with_v1_optimizers_test.py | 197 root_trackable.save_counter # pylint: disable=pointless-statement 207 self.evaluate(state_ops.assign(root_trackable.save_counter, 3)) 214 self.assertAllEqual(1, self.evaluate(root_trackable.save_counter)) 234 self.assertAllEqual(1, self.evaluate(on_create_root.save_counter)) 382 session.run(root.save_counter)) 415 self.evaluate(root.save_counter)) 459 self.evaluate(root.save_counter))
|
D | tracking_util_test.py | 211 self.assertFalse(root_trackable.save_counter.trainable) 221 self.evaluate(state_ops.assign(root_trackable.save_counter, 3)) 229 self.assertAllEqual(1, self.evaluate(root_trackable.save_counter)) 244 self.assertAllEqual(1, self.evaluate(on_create_root.save_counter)) 331 session.run(root.save_counter)) 366 self.evaluate(root.save_counter)) 433 self.evaluate(root.save_counter))
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.train.-checkpoint.pbtxt | 8 name: "save_counter"
|
/external/tensorflow/tensorflow/tools/api/golden/v2/ |
D | tensorflow.train.-checkpoint.pbtxt | 8 name: "save_counter"
|