PiperOrigin-RevId: 216751041

In control flow functionalization, mark generated If/While with outside compilation attribute if necessary.

PiperOrigin-RevId: 216750679

PiperOrigin-RevId: 216748472

Validate the control flow information after subgraph rewrite, since the rewrite may remove some special nodes (e.g. TPUReplicateMetadata) which break the rule.

PiperOrigin-RevId: 216746513

PiperOrigin-RevId: 216746022

PiperOrigin-RevId: 216745236

PiperOrigin-RevId: 216738986

- This CL intruduces input/output alias config in HLO module that allows any HLO pass to configure it. Once the alias_config is set, each backend needs to follow the contract during execution time to make sure the input and output are indeed aliased.

- Copy insertion / buffer assignment and alias analysis has been updated to correctly honor the config and avoid any possible liveness interference.

PiperOrigin-RevId: 216737975

The evaluator would crash when encountering an instruction which produced a
TOKEN shape.

PiperOrigin-RevId: 216737937

PiperOrigin-RevId: 216736399

PiperOrigin-RevId: 216736164

This quantization is configured so that signed integers can be used for inference without biasing, so narrow_range=false implies the negative limit is slightly larger than the positive limit, as is the case for signed two's-complement representation.

PiperOrigin-RevId: 216734571

PiperOrigin-RevId: 216734498

PiperOrigin-RevId: 216732038

PiperOrigin-RevId: 216728781

Modify converted_call to accept a separate argument when the callable is an attribute, representing the attribute's owner.

This gives converted_call a chance to override the attribute lookup, and in turn allows working around the limitation `super` that prevents it from resolving dynamic attributes.

PiperOrigin-RevId: 216727687

Also fix the DotLayout test, it would pass even when commenting out the dot specific logic in GpuLayoutAssignment.

PiperOrigin-RevId: 216726796

PiperOrigin-RevId: 216725062

patch command can exceed the 10s timeout limit on some slow machines.

Fixes #22761

PiperOrigin-RevId: 216721485

Use the functional operator pattern for logical expressions. The current implementation relied on the TF import, which is now gone.

PiperOrigin-RevId: 216719675

PiperOrigin-RevId: 216718180

PiperOrigin-RevId: 216714487

Also fix the initial_state and constants param in the eager context.

PiperOrigin-RevId: 216712447

PiperOrigin-RevId: 216706771

PiperOrigin-RevId: 216706627

PiperOrigin-RevId: 216705389

PiperOrigin-RevId: 216705354

PiperOrigin-RevId: 216697820

PiperOrigin-RevId: 216697425

PiperOrigin-RevId: 216691593

PiperOrigin-RevId: 216688700

PiperOrigin-RevId: 216686849

PiperOrigin-RevId: 216686136

We are working on disallowing xla::Add on PRED types because they can be
confusing. This is a prerequiste for that to not fail over in the new
shape check.

PiperOrigin-RevId: 216685015

Wonder how that could have ever let this CL to go in in the first place.

Automated rollback of commit 905deeaa

PiperOrigin-RevId: 216683052

PiperOrigin-RevId: 216663386

PiperOrigin-RevId: 216639755

PiperOrigin-RevId: 216638239

PiperOrigin-RevId: 216633097

PiperOrigin-RevId: 216632480