xla

41d3e781 - handle all_reduce in place for eager mode