PiperOrigin-RevId: 646609853

Also change an `if_static` to an `if_google` to fix JAX builds

PiperOrigin-RevId: 646622609

Tested both libtensorflowlite.so and libtensorflowlite_jni.so to ensure both
libraries are 16k ELF aligned with this change:

$ objdump -p bazel-bin/tensorflow/lite/libtensorflowlite.so | grep LOAD | awk '{ print $1 " " $NF }'
LOAD 2**14
LOAD 2**14

$ objdump -p bazel-bin/tensorflow/lite/java/libtensorflowlite_jni.so | grep LOAD | awk '{ print $1 " " $NF }'
LOAD 2**14
LOAD 2**14

PiperOrigin-RevId: 646629366

PiperOrigin-RevId: 646631043

PiperOrigin-RevId: 646633442

… `package_groups` and replace Copybara rules

PiperOrigin-RevId: 646633709

…test to make sure crashing Triton configurations are actually skipped and to guard against breaking it.

PiperOrigin-RevId: 646663340

…to the kernel

PiperOrigin-RevId: 646664927

…ces.

This CL ensures that AllReduceSimplifier can simplify trivial all-reduces (an all-reduce where each subgroup is formed of a single participant) that are not necessarily cross replica (for example a cross partition all-reduce). We only simplify non cross replica all-reduce when the module is SPMD.

FUTURE_COPYBARA_INTEGRATE_REVIEW=openxla/xla#14073 from apivovarov:select_compare_algsimp 6fe68d7319b272ff041b67e038359540cddda489
PiperOrigin-RevId: 644267887