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"
|