You signed in with another tab or window. Reload to refresh your session.You signed out in another tab or window. Reload to refresh your session.You switched accounts on another tab or window. Reload to refresh your session.Dismiss alert
Cherrypick DTensor docstring fix for 2.9 release. (#16434)
* 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
0 commit comments