Less restrictive fallback logic (keras-team#17591)

Co-authored-by: Eugene Kuznetsov <[email protected]>

Less restrictive fallback logic (keras-team#17591)

Co-authored-by: Eugene Kuznetsov <[email protected]>

Merge pull request keras-team#17225 from lgeiger:fix-mixed-precision-…

…ema (keras-team#17226)

PiperOrigin-RevId: 486999108

Co-authored-by: TensorFlower Gardener <[email protected]>

Merge pull request keras-team#17225 from lgeiger:fix-mixed-precision-…

…ema (keras-team#17226)

PiperOrigin-RevId: 486999108

Co-authored-by: TensorFlower Gardener <[email protected]>

Cherrypick for cl/482011499: Throw error on deprecated fields. (keras…

…-team#17179)

* Throw error on deprecated fields.

PiperOrigin-RevId: 482011499

* copyedits

Co-authored-by: Chen Qian <[email protected]>
Co-authored-by: François Chollet <[email protected]>

Cherrypick 2 more changes for the optimizer docstring fix. (keras-tea…

…m#17165)

* keras: fix typo in keras optimizers migration recommendation

  tf.keras.optimizer.legacy -> tf.keras.optimizers.legacy

PiperOrigin-RevId: 481720094

* Keras: Fix docstring for tf.keras.optimizer. -> tf.keras.optimizers.

PiperOrigin-RevId: 481721074

Co-authored-by: Michael Chinen <[email protected]>
Co-authored-by: A. Unique TensorFlower <[email protected]>

Add `is_legacy_optimizer` to optimizer config to keep saving/loading …

…consistent. (keras-team#16856)

PiperOrigin-RevId: 463928027

Co-authored-by: Chen Qian <[email protected]>

Add `is_legacy_optimizer` to optimizer config to keep saving/loading …

…consistent. (keras-team#16856)

PiperOrigin-RevId: 463928027

Co-authored-by: Chen Qian <[email protected]>

Cherrypick ImageAugmetation related change to R2.9 (keras-team#16454)

* Enable the keras dtensor API in OSS.

PiperOrigin-RevId: 438858608

* Switching learning/brain dependency to OSS compatible test_util

This is one test file failing, due to the monkey patching happens in the dtensor.init(), and I will need to dig more about the root cause (probably due to patching tf.Variable with DVariable, and cause logic difference for instance type checking.)

PiperOrigin-RevId: 439676157

* Update the docstring for keras.dtensor components.

1. Add docstring for LayoutMap.
2. Hide certain methods for keras.dtensor.optimizers.

PiperOrigin-RevId: 442651549

* add no-op label augmentation to RandomContrast

PiperOrigin-RevId: 439938098

* Add augment_label to RandomBrightness

PiperOrigin-RevId: 443122383

* Create augment_label method:
```
  def augment_label(self, label, transformation=None):
    return label
```
on remaining image augmentation layers

PiperOrigin-RevId: 443446864

* Revert "Create augment_label method:"

This reverts commit 1694cb4.

Co-authored-by: Luke Wood <[email protected]>

Cherrypick ImageAugmetation related change to R2.9 (keras-team#16454)

* Enable the keras dtensor API in OSS.

PiperOrigin-RevId: 438858608

* Switching learning/brain dependency to OSS compatible test_util

This is one test file failing, due to the monkey patching happens in the dtensor.init(), and I will need to dig more about the root cause (probably due to patching tf.Variable with DVariable, and cause logic difference for instance type checking.)

PiperOrigin-RevId: 439676157

* Update the docstring for keras.dtensor components.

1. Add docstring for LayoutMap.
2. Hide certain methods for keras.dtensor.optimizers.

PiperOrigin-RevId: 442651549

* add no-op label augmentation to RandomContrast

PiperOrigin-RevId: 439938098

* Add augment_label to RandomBrightness

PiperOrigin-RevId: 443122383

* Create augment_label method:
```
  def augment_label(self, label, transformation=None):
    return label
```
on remaining image augmentation layers

PiperOrigin-RevId: 443446864

* Revert "Create augment_label method:"

This reverts commit 1694cb4.

Co-authored-by: Luke Wood <[email protected]>