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 TensorFlow: Upstream a batch of changes to git. 2015-11-09 18:42:53 -08:00
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 fix docs for rest training module 2019-04-03 09:16:31 +02:00
adagrad_da_test.py Edited comments for test script 2019-08-07 19:35:05 +10:00
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 Add @run_deprecated_v1 annotation to tests failing in v2 2018-11-29 15:43:25 -08:00
basic_loops.py Apply tf1->tf2 name replaces to doc-strings and comments in tensorflow. 2019-04-18 17:19:27 -07:00
basic_session_run_hooks_test.py (1/2) Add option to not export Graph protos when training the Estimator. 2019-10-10 12:40:44 -07:00
basic_session_run_hooks.py Minor improved log statements for CheckpointSaverHook. 2020-02-01 03:24:01 -08:00
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 Small adjustments on import spacing. 2019-12-19 12:27:39 -08:00
checkpoint_ops.py Expose an axis argument for VocabInfo, which allows for warm-starting of the second axis of Tensors through tf.train.warm_start. Note that the underlying initializer already has this functionality (for example, for output layers). 2018-09-05 15:42:44 -07:00
checkpoint_state.proto Add tf.contrib.checkpoint.CheckpointManager for deleting old checkpoints 2018-08-13 16:53:59 -07:00
checkpoint_utils_test.py Small adjustments on import spacing. 2019-12-19 12:27:39 -08:00
checkpoint_utils.py Small adjustments on import spacing. 2019-12-19 12:27:39 -08:00
coordinator_test.py Fix a typo in coordinator_test.py. threads[1:1] is the empty array, so we were't 2019-10-01 19:50:42 -07:00
coordinator.py Remove train.LooperThread from TF 2.0 API. 2018-11-26 15:26:01 -08:00
device_setter_test.py Add @run_deprecated_v1 annotation to tests failing in v2 2018-11-29 15:43:25 -08:00
device_setter.py Tensorforest: adding resource to the list of ops to be created on PS 2019-08-28 13:54:35 -07:00
distribute.py Move distribute.py, distribution_strategy_context.py, and device_util.py 2018-11-25 18:06:53 -08:00
distribution_strategy_context.py Move distribute.py, distribution_strategy_context.py, and device_util.py 2018-11-25 18:06:53 -08:00
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 Add @run_deprecated_v1 annotation to tests failing in v2 2018-11-29 15:43:25 -08:00
gradient_descent.py Deprecate tf.train._ classes in TF 2.0 API in favor of tf.keras.optimizers._ ones. 2018-11-28 08:33:00 -08:00
input_test.py Replace list comprehension with generator expressions. 2019-12-16 12:26:12 -08:00
input.py Fixed cases where tf.TensorShape was constructed with float dimensions 2019-10-14 20:53:35 -07:00
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 Annotate tests as @run_v1_only 2018-12-07 17:41:47 -08:00
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 Merge pull request #24796 from mrTsjolder:unify_doccitations 2019-11-19 18:24:34 -08:00
monitored_session_test.py minor spelling tweaks 2020-02-11 15:09:21 +09:00
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 Export the bfloat16 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-12-02 16:17:38 -08:00
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 Add @run_deprecated_v1 annotation to tests failing in v2 2018-11-29 15:43:25 -08:00
proximal_adagrad.py first steps in cleaning doc citations 2019-04-03 09:16:31 +02:00
proximal_gradient_descent_test.py Add @run_deprecated_v1 annotation to tests failing in v2 2018-11-29 15:43:25 -08:00
proximal_gradient_descent.py first steps in cleaning doc citations 2019-04-03 09:16:31 +02:00
py_checkpoint_reader.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
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 Export the quantize_training 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-10-21 17:20:05 -07:00
queue_runner_impl.py Apply tf1->tf2 name replaces to doc-strings and comments in tensorflow. 2019-04-18 17:19:27 -07:00
queue_runner_test.py Update @run_v1_only annotation for symbols removed from v2 2018-12-17 11:09:45 -08:00
queue_runner.py Removing remove_undocumented calls from tensorflow/python. 2018-04-25 11:57:31 -07:00
rmsprop_test.py Add @run_deprecated_v1 annotation to tests failing in v2 2018-11-29 15:43:25 -08:00
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 Replaced deprecated tf.create_partitioned_variables with tf.get_variable 2018-11-27 03:03:27 -08:00
saver_large_variable_test.py Remove hourglass imports from even more tests 2016-12-16 18:26:25 -08:00
saver_test_utils.py eager: Rename in_eager_mode to executing_eagerly and get rid of in_graph_mode. 2018-03-07 12:10:42 -08:00
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 Add @run_deprecated_v1 annotation to tests failing in v2 2018-11-29 15:43:25 -08:00
server_lib_same_variables_clear_container_test.py Add @run_deprecated_v1 annotation to tests failing in v2 2018-11-29 15:43:25 -08:00
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 Annotate tests as @run_v1_only 2018-12-07 17:41:47 -08:00
server_lib_sparse_job_test.py Add @run_deprecated_v1 annotation to tests failing in v2 2018-11-29 15:43:25 -08:00
server_lib_test.py Apply tf1->tf2 name replaces to doc-strings and comments in tensorflow. 2019-04-18 17:19:27 -07:00
server_lib.py add case when errors is garbage collected 2020-04-24 09:38:42 +08:00
session_manager_test.py Add SavedModelEstimator support for saved models with assets. 2019-09-23 21:04:04 -07:00
session_manager.py Small adjustments on import spacing. 2019-12-19 12:27:39 -08:00
session_run_hook.py Apply tf1->tf2 name replaces to doc-strings and comments in tensorflow. 2019-04-18 17:19:27 -07:00
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 minor spelling tweaks 2020-02-11 15:09:21 +09:00
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 Apply run_deprecated_v1 to entire test class 2018-12-18 02:31:34 -08:00
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 Add pylint check for W0611 unused-import in ci_sanity.sh and fix existing pylint errors. 2018-02-06 22:09:14 -08:00
training_util_test.py Annotate tests as @run_v1_only 2018-12-07 17:41:47 -08:00
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 Small adjustments on import spacing. 2019-12-19 12:27:39 -08:00
warm_starting_util.py fix: Resolving incorrect value change error 2020-04-07 20:06:23 +09:00