Yuanzhong Xu 19624f9650 [XLA:SPMD] Add API for the experimental mechanism to mix auto and manual partitioning ... There are two ops added: XlaSpmdFullToShardShape: casts full shape (to be auto partitioned) to shard shape, which can then be consumed by manually partitioned code. XlaSpmdShardToFullShape: casts shard shape (manually partitioned) to full shape, which will be consumed by ops auto-partitioned by SPMD. PiperOrigin-RevId: 309845623 Change-Id: Ic056f2965c04a2357c2bf63d642eeafdbaa19b18		2020-05-04 16:57:03 -07:00
..
BUILD	…
xla_ops.cc	[XLA:SPMD] Add API for the experimental mechanism to mix auto and manual partitioning	2020-05-04 16:57:03 -07:00