Skip to content

Fix double-compilation in train_step by matching input sharding.#4174

Open
igorts-git wants to merge 1 commit into
mainfrom
igorts/b440499679-train_step-hlo
Open

Fix double-compilation in train_step by matching input sharding.#4174
igorts-git wants to merge 1 commit into
mainfrom
igorts/b440499679-train_step-hlo

Fix double-compilation in train_step by propagating input sharding to…

1463b83
Select commit
Loading
Failed to load commit list.
Google CLA / cla/google succeeded Jun 16, 2026 in 6s

✅ All contributors are covered under a CLA with Google

See https://cla.developers.google.com/ for more info about Google's Contributor License Agreement (CLA).

ℹ️ Googlers: Go here to view more details and manage scans for this pull request.

Details

The following contributors were found for this pull request:

1463b83 Author: @igorts-git <ig***s​@google.com>

(Only the first commit for a unique contributor is listed.)