STT-tensorflow/tensorflow/python/training
Ken Franko b8694e39d8 Correctly set the experimental_io_device when restoring variable from a checkpoint.
PiperOrigin-RevId: 320222381
Change-Id: I30187c7777ab8056e48004ef5e4ae747edc32227
2020-07-13 15:50:48 -07:00
..
experimental Fork the keras related MP test to keras folder. 2020-06-12 15:12:04 -07:00
saving Add SaveableObjects to SavedModel. 2020-07-01 14:28:41 -07:00
tracking Correctly set the experimental_io_device when restoring variable from a checkpoint. 2020-07-13 15:50:48 -07:00
__init__.py
adadelta_test.py Remove reset_test argument from test_util.run_in_graph_and_eager_modes 2020-04-30 15:09:55 -07:00
adadelta.py
adagrad_da_test.py
adagrad_da.py Merge pull request #24796 from mrTsjolder:unify_doccitations 2019-11-19 18:24:34 -08:00
adagrad_test.py Remove reset_test argument from test_util.run_in_graph_and_eager_modes 2020-04-30 15:09:55 -07:00
adagrad.py Merge pull request #24796 from mrTsjolder:unify_doccitations 2019-11-19 18:24:34 -08:00
adam_test.py Remove reset_test argument from test_util.run_in_graph_and_eager_modes 2020-04-30 15:09:55 -07:00
adam.py Fix documentation compatibility tag formatting. 2020-05-07 16:46:21 -07:00
basic_loops_test.py
basic_loops.py
basic_session_run_hooks_test.py
basic_session_run_hooks.py
checkpoint_management_test.py Extend tf.train.CheckpointManager functionality: 2020-02-12 19:37:30 -08:00
checkpoint_management.py Add more information in CheckpointManager's summary line. 2020-04-17 21:20:09 -07:00
checkpoint_ops_test.py
checkpoint_ops.py
checkpoint_state.proto Add tf.contrib.checkpoint.CheckpointManager for deleting old checkpoints 2018-08-13 16:53:59 -07:00
checkpoint_utils_test.py
checkpoint_utils.py
coordinator_test.py
coordinator.py
device_setter_test.py
device_setter.py
distribute.py
distribution_strategy_context.py
evaluation_test.py Update evaluation_test to not rely on Keras metrics. 2020-06-11 10:57:06 -07:00
evaluation.py Merge pull request #31833 from cocopambag:master 2019-10-09 15:28:55 -07:00
ftrl_test.py Switch tests of the v1 ftrl optimizer (which is graph only) to use ops.Graph.as_default() explicitly instead of relying on the run_deprecated_v1 decorator (which we're trying to avoid) 2020-06-10 14:25:18 -07:00
ftrl.py Remove colocate_with function from FTRL optimizer for use with DistributionStrategy. 2019-12-23 22:01:18 -08:00
gradient_descent_test.py
gradient_descent.py
input_test.py
input.py
learning_rate_decay.py Move the learning_rate_decay code to keras to break the reverse dependency. 2020-03-30 11:06:36 -07:00
localhost_cluster_performance_test.py
momentum_test.py Remove reset_test argument from test_util.run_in_graph_and_eager_modes 2020-04-30 15:09:55 -07:00
momentum.py
monitored_session_test.py
monitored_session.py Change the logging from warning to error when there is an exception at session close. 2020-05-08 11:51:34 -07:00
moving_averages_test.py
moving_averages.py In assign_moving_average, call update_fn instead of strategy.extended.update(var, update_fn) when in update context. 2020-06-10 17:52:55 -07:00
optimizer_test.py Use same var key in _create_slots/get_slot in V1 optimizer 2020-03-18 13:49:52 -07:00
optimizer.py Rename _get_closest to more accurately reflect what it does. 2020-05-18 14:37:31 -07:00
proximal_adagrad_test.py
proximal_adagrad.py first steps in cleaning doc citations 2019-04-03 09:16:31 +02:00
proximal_gradient_descent_test.py
proximal_gradient_descent.py
py_checkpoint_reader.py
quantize_training_test.py Update v1 only test with proper reason. 2020-06-04 12:39:12 -07:00
quantize_training_wrapper.cc [Build cleanup] Split "core_cpu_impl" into fine-grained targets (4/n). 2020-04-28 09:53:59 -07:00
quantize_training.py
queue_runner_impl.py
queue_runner_test.py
queue_runner.py
rmsprop_test.py
rmsprop.py Correct the formulation in the description of the centered rmsprop 2020-03-09 11:11:28 -07:00
saver_large_partitioned_variable_test.py
saver_large_variable_test.py
saver_test_utils.py
saver_test.py Fork the keras related saver_test to keras. 2020-06-15 09:02:54 -07:00
saver.py Merge pull request #36388 from rahul003:s3_skip_temp 2020-02-26 14:49:31 -08:00
server_lib_multiple_containers_test.py
server_lib_same_variables_clear_container_test.py
server_lib_same_variables_clear_test.py Add @run_deprecated_v1 annotation to tests failing in v2 2018-11-29 15:43:25 -08:00
server_lib_same_variables_no_clear_test.py
server_lib_sparse_job_test.py
server_lib_test.py
server_lib.py add case when errors is garbage collected 2020-04-24 09:38:42 +08:00
session_manager_test.py
session_manager.py
session_run_hook.py
slot_creator_test.py Merge pull request #30018 from candyzone:master 2019-09-05 17:47:46 -07:00
slot_creator.py Merge pull request #30018 from candyzone:master 2019-09-05 17:47:46 -07:00
summary_io.py Apply tf1->tf2 name replaces to doc-strings and comments in tensorflow. 2019-04-18 17:19:27 -07:00
supervisor_test.py Update v1 only test with proper reason. 2020-06-01 13:59:27 -07:00
supervisor.py Fix error message in python/training/supervisor.py 2019-11-21 15:16:16 +08:00
sync_replicas_optimizer_test.py
sync_replicas_optimizer.py Merge pull request #34985 from kiszk:spelling_tweaks_python 2020-02-12 13:41:49 -08:00
tensorboard_logging_test.py
tensorboard_logging.py Add a recursive import before running doctest. 2020-06-01 21:13:25 -07:00
training_ops_test.py Added CPU and GPU implementations of a modified version of FTRL, triggered using an attribute, that multiplies the "linear" accumulator by the learning rate. This enables a learning rate of zero to work correctly. 2020-04-07 12:11:41 -07:00
training_ops.py
training_util_test.py
training_util.py Apply tf1->tf2 name replaces to doc-strings and comments in tensorflow. 2019-04-18 17:19:27 -07:00
training.py Export the checkpoint reader classes and functions from C++ to Python with pybind11 instead of swig. This is part of a larger effort to deprecate swig and eventually with modularization break pywrap_tensorflow into smaller components. It will also make exporting C++ ops to Python significantly easier. XLA is using the pybind11 macros already. Please refer to https://github.com/tensorflow/community/blob/master/rfcs/20190208-pybind11.md for more information. 2019-11-07 09:52:50 -08:00
warm_starting_util_test.py
warm_starting_util.py fix: Resolving incorrect value change error 2020-04-07 20:06:23 +09:00