          if skip_check:
              # TODO remove this hacky fix to disable mixed precision compatibility testing if/when
              # fixed upstream.
              # pylint:disable=import-outside-toplevel,protected-access
              from tensorflow.python.keras.mixed_precision.experimental import \
                  device_compatibility_check
              logger.debug("Overriding tensorflow _logged_compatibility_check parameter. Initial "
                           "value: %s", device_compatibility_check._logged_compatibility_check)
              device_compatibility_check._logged_compatibility_check = True
              logger.debug("New value: %s", device_compatibility_check._logged_compatibility_check)
 
</s> remove         self._mixed_precision.set_policy(policy)
 </s> add         self._mixed_precision.set_global_policy(policy) </s> remove from tensorflow.python import errors_impl as tf_errors  # pylint:disable=no-name-in-module
 </s> add from tensorflow.python.framework import errors_impl as tf_errors  # pylint:disable=no-name-in-module </s> remove         max_ver = 2.2
 </s> add         max_ver = 2.4 </s> remove from tensorflow.python import errors_impl as tf_errors  # pylint:disable=no-name-in-module
 </s> add from tensorflow.python.framework import errors_impl as tf_errors  # pylint:disable=no-name-in-module </s> remove from tensorflow.python import errors_impl as tf_errors  # pylint:disable=no-name-in-module
 </s> add from tensorflow.python.framework import errors_impl as tf_errors  # pylint:disable=no-name-in-module
            from tensorflow.python.keras.mixed_precision import \
          if skip_check:
              # TODO remove this hacky fix to disable mixed precision compatibility testing if/when
              # fixed upstream.
              # pylint:disable=import-outside-toplevel,protected-access
            from tensorflow.python.keras.mixed_precision import \
                  device_compatibility_check
              logger.debug("Overriding tensorflow _logged_compatibility_check parameter. Initial "
                           "value: %s", device_compatibility_check._logged_compatibility_check)
              device_compatibility_check._logged_compatibility_check = True
              logger.debug("New value: %s", device_compatibility_check._logged_compatibility_check)
keep keep keep keep replace keep keep keep keep keep
 Update tensorflow to 2.4.0rc4 (#1086)

* Update tensorflow to 2.4.0rc1

* Update requirements_nvidia.txt

* Update requirements_nvidia.txt 
plugins/train/model/_base.py