Yuanzhong Xu 7c096232b2 [XLA] Avoid reshape to R1 in NormalFloatingPointDistribution ...

This refactors existing shape split code and reuses it for NormalFloatingPointDistribution

PiperOrigin-RevId: 347643318
Change-Id: Ie3d3c0145cf6a275a5f678ee4258f88c7de18a5a

2020-12-15 11:13:11 -08:00

..

lib

[XLA] Avoid reshape to R1 in NormalFloatingPointDistribution

2020-12-15 11:13:11 -08:00

BUILD

Move CreateModuleConfig to a new hlo_module_util header.

2020-12-14 15:14:29 -08:00

client_library.cc

…

client_library.h

…

client.cc

xla directory resolutions

2020-07-26 22:14:33 +00:00

client.h

…

compile_only_client.cc

…

compile_only_client.h

…

executable_build_options.cc

Move CreateModuleConfig to a new hlo_module_util header.

2020-12-14 15:14:29 -08:00

executable_build_options.h

[PJRT] Parallelize GPU compilation.

2020-12-15 06:20:58 -08:00

global_data.cc

…

global_data.h

…

local_client.cc

Remove platform field from shaped buffer.

2020-10-30 14:26:55 -07:00

local_client.h

…

padding_test.cc

…

padding.cc

…

padding.h

…

sharding_builder.cc

[XLA:SPMD] Minor fixes and utils for manual sharding

2020-12-01 13:37:36 -08:00

sharding_builder.h

[XLA:SPMD] Minor fixes and utils for manual sharding

2020-12-01 13:37:36 -08:00

xla_builder_test.cc

Speed up Shape creation by avoiding unnecessary validations.

2020-11-22 20:40:07 -08:00

xla_builder.cc

[XLA] Add creation_pass_id and dummy op_names to OpMetadata.

2020-12-11 17:48:13 -08:00

xla_builder.h

[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

xla_computation.cc

…

xla_computation.h

…