diff options
author | 2017-10-05 12:55:19 -0700 | |
---|---|---|
committer | 2017-10-05 13:00:01 -0700 | |
commit | b31c03565e18fef7ab4539032dd5c69a94487a05 (patch) | |
tree | 737dc9ea986b7b02126bce3d78e054cc749837b4 /tensorflow/python/training/training.py | |
parent | 4bf27f8d4acee2cb8df27427668bddc92137e2ef (diff) |
Move profiler hook from contrib to core.
PiperOrigin-RevId: 171194291
Diffstat (limited to 'tensorflow/python/training/training.py')
-rw-r--r-- | tensorflow/python/training/training.py | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tensorflow/python/training/training.py b/tensorflow/python/training/training.py index e2a7b28e2b..741dddc991 100644 --- a/tensorflow/python/training/training.py +++ b/tensorflow/python/training/training.py @@ -77,6 +77,7 @@ See the @{$python/train} guide. @@GlobalStepWaiterHook @@FinalOpsHook @@FeedFnHook +@@ProfilerHook @@SecondOrStepTimer @@global_step @@basic_train_loop @@ -145,6 +146,7 @@ from tensorflow.python.training.basic_session_run_hooks import SummarySaverHook from tensorflow.python.training.basic_session_run_hooks import GlobalStepWaiterHook from tensorflow.python.training.basic_session_run_hooks import FinalOpsHook from tensorflow.python.training.basic_session_run_hooks import FeedFnHook +from tensorflow.python.training.basic_session_run_hooks import ProfilerHook from tensorflow.python.training.basic_loops import basic_train_loop from tensorflow.python.training.checkpoint_utils import init_from_checkpoint from tensorflow.python.training.checkpoint_utils import list_variables |