jax
169a123e - Add _XlaShardingV2 to tf.XlaShardOp and use it for tf2xla lowering.

Commit

218 days ago

Add _XlaShardingV2 to tf.XlaShardOp and use it for tf2xla lowering. PiperOrigin-RevId: 776237944

References

Author

bixia1

Committer

Google-ML-Automation

Parents