Skip to content

Cherrypick DTensor docstring fix for 2.9 release.#16434

Merged
qlzh727 merged 4 commits intokeras-team:r2.9from
qlzh727:r2.9
Apr 18, 2022
Merged

Cherrypick DTensor docstring fix for 2.9 release.#16434
qlzh727 merged 4 commits intokeras-team:r2.9from
qlzh727:r2.9

Conversation

@qlzh727
Copy link
Copy Markdown
Member

@qlzh727 qlzh727 commented Apr 18, 2022

Piper cl: 442651549

qlzh727 added 4 commits April 7, 2022 15:55
PiperOrigin-RevId: 438858608
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
1. Add docstring for LayoutMap.
2. Hide certain methods for keras.dtensor.optimizers.

PiperOrigin-RevId: 442651549
@qlzh727 qlzh727 requested a review from fchollet April 18, 2022 22:51
@google-ml-butler google-ml-butler bot added the keras-team-review-pending Pending review by a Keras team member. label Apr 18, 2022
@qlzh727 qlzh727 removed the keras-team-review-pending Pending review by a Keras team member. label Apr 18, 2022
@qlzh727 qlzh727 merged commit 970c060 into keras-team:r2.9 Apr 18, 2022
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment

Labels

Projects

None yet

Development

Successfully merging this pull request may close these issues.

2 participants