Searched refs:save_counter (Results 1 – 10 of 10) sorted by relevance
/external/tensorflow/tensorflow/python/training/ |
D | checkpoint_management.py | 671 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
|
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 | 1304 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)
|
D | util_with_v1_optimizers_test.py | 202 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))
|
D | util_test.py | 409 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/ |
D | checkpointable_utils_test.py | 220 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/ |
D | tensorflow.train.-checkpoint.pbtxt | 8 name: "save_counter"
|
/external/tensorflow/tensorflow/tools/api/golden/v1/ |
D | tensorflow.train.-checkpoint.pbtxt | 8 name: "save_counter"
|
/external/tensorflow/tensorflow/examples/tf2_showcase/ |
D | mnist.py | 225 (checkpoint.save_counter.numpy() + 1,
|
/external/tensorflow/tensorflow/contrib/eager/python/examples/gan/ |
D | mnist.py | 306 (checkpoint.save_counter.numpy(),
|