Yuanzhong Xu 97b414dd46 [XLA:SPMD] Add partial sharding API to SPMD and bridge support ...

PiperOrigin-RevId: 326348623
Change-Id: I32e94f708ff7c13aa17fabac645100f178c2e1be

2020-08-12 17:37:57 -07:00

..

BUILD

Expose XlaOptions as an argument to tpu.rewrite

2020-07-08 15:56:29 -07:00

xla_sharding.py

[XLA:SPMD] Add partial sharding API to SPMD and bridge support

2020-08-12 17:37:57 -07:00