flax
115dcf58 - add graph_updates argument in shard_map

Commit

149 days ago

add graph_updates argument in shard_map PiperOrigin-RevId: 882638167

References

#5319 - add graph_updates argument in shard_map

Author

Cristian Garcia

Committer

a-googler

a-googler

Parents

Loading