A. Unique TensorFlower 629a241b55 [XLA:SPMD] Apply all to all when resharding from/to partial replicate at some ...

special cases.

PiperOrigin-RevId: 348091566
Change-Id: I178f5dc9fe83cb9bcf45a87998cb755691935b4f

2020-12-17 14:19:58 -08:00

..

BUILD

[XLA:SPMD] Add basic support for SPMD FFT.

2020-10-21 11:19:54 -07:00

convolution_handler.cc

[XLA] Add an optional preferred_element_type argument to Dot/Conv builder methods to enable generating HLOs that has wider accumulation type than default shape inference result.

2020-11-20 13:16:56 -08:00

convolution_handler.h

[XLA:SPMD] Support partition convlution input in both spatial dimension and

2020-09-18 23:54:09 -07:00

dot_handler.cc

[XLA:SPMD] Copy rhs when lhs and rhs of dot are identical.

2020-11-20 21:02:49 -08:00

fft_handler.cc

[XLA:SPMD] Add basic support for SPMD FFT.

2020-10-21 11:19:54 -07:00

schedule_aware_all_gather_cse.cc

[XLA] Fix a live-range heuristic in all-gather CSE

2020-09-13 22:45:24 -07:00

schedule_aware_all_gather_cse.h

[XLA:SPMD] Schedule-aware all-gather CSE for SPMD

2020-08-19 15:58:14 -07:00

spmd_partitioner_test.cc

[XLA:SPMD] Apply all to all when resharding from/to partial replicate at some

2020-12-17 14:19:58 -08:00

spmd_partitioner_util.cc

Mask operands for unevenly partitioned contracting dims.

2020-11-05 23:53:13 -08:00

spmd_partitioner_util.h

[XLA:SPMD] Support Scatter in Partial Replicate.

2020-08-28 15:57:17 -07:00

spmd_partitioner.cc

[XLA:SPMD] Apply all to all when resharding from/to partial replicate at some

2020-12-17 14:19:58 -08:00

spmd_partitioner.h

[XLA:SPMD] Apply all to all when resharding from/to partial replicate at some

2020-12-17 14:19:58 -08:00