xla
Implement `NamedSharding::ToTileAssignment` converting `NamedSharding` to `TileAssignment` representation, required for implementing methods involving tile based indexing `TileIndexForDevice`, `TileOffsetForDevice` etc, e.g: cl/855697055
#36410

Open

Implement `NamedSharding::ToTileAssignment` converting `NamedSharding` to `TileAssignment` representation, required for implementing methods involving tile based indexing `TileIndexForDevice`, `TileOffsetForDevice` etc, e.g: cl/855697055 #36410

copybara-service wants to merge 1 commit into main from test_855697056

Implement `NamedSharding::ToTileAssignment` converting `NamedSharding…

e5278cf3

copybara-service assigned

KanishAnand 19 hours ago

Reviewers

No reviews

Assignees

KanishAnand

Labels

None yet

Milestone

No milestone