Skip to content

v2.9.0-rc2

v2.9.0-rc2: Cherrypick ImageAugmetation related change to R2.9 (#16454)
@qlzh727 qlzh727 tagged this 22 Apr 17:34 
* 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 1694cb4ea0ca084c49a178afe45bbee4379a2cc4.

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