[determinism] Add segment reduction op exceptions for GPU determinism#47772

Merged

copybara-service[bot] merged 4 commits intotensorflow:masterfrom

duncanriach:segment-sum-nond9m-exceptions

Mar 18, 2021

Commits on Mar 12, 2021

[determinism] Add segment reduction op exceptions
duncanriach
committed

Commits on Mar 15, 2021

[determinism] Disable XLA auto-jit for segment reduction d9m-unimplemented tests
duncanriach
committed

Commits on Mar 16, 2021

[determinism] Address review, step 1, on PR 47772
duncanriach
committed

Commits on Mar 17, 2021

[determinism] Fix buildifier error
duncanriach
committed