From 6fd3e5e217a5ef4656c40bf0171a025d7b9d9183 Mon Sep 17 00:00:00 2001 From: Priya Gupta Date: Tue, 7 Jul 2020 22:56:56 -0700 Subject: [PATCH] Remove ReplicaContext.__enter__ and __exit__ methods from the docs since users are not expected to use them directly. PiperOrigin-RevId: 320129981 Change-Id: Ifd3ec638549de95c01147560260b8ecf7f17e29d --- tensorflow/python/distribute/distribute_lib.py | 2 ++ 1 file changed, 2 insertions(+) diff --git a/tensorflow/python/distribute/distribute_lib.py b/tensorflow/python/distribute/distribute_lib.py index 4a57628d4c7..78a199fe782 100644 --- a/tensorflow/python/distribute/distribute_lib.py +++ b/tensorflow/python/distribute/distribute_lib.py @@ -2728,6 +2728,7 @@ class ReplicaContext(object): self._replica_id_in_sync_group = replica_id_in_sync_group self._summary_recording_distribution_strategy = None + @doc_controls.do_not_generate_docs def __enter__(self): _push_per_thread_mode(self._thread_context) @@ -2740,6 +2741,7 @@ class ReplicaContext(object): summary_state.is_recording_distribution_strategy) summary_state.is_recording_distribution_strategy = replica_id_is_zero + @doc_controls.do_not_generate_docs def __exit__(self, exception_type, exception_value, traceback): summary_state = summary_ops_v2._summary_state # pylint: disable=protected-access summary_state.is_recording_distribution_strategy = (