[XLA] Fix dynamic expression edge cases and batch bucketing - #55

Open

stevenvar wants to merge 17 commits into

for-serving-2.20from

steven.fix-for-serving-review-findings

Commits on Aug 14, 2026

Add dynamic size expression tests and fix propagation gaps (#27)
stevenvar
committed
7c85598
Support max and conditional dynamic shape expressions (#4)
stevenvar
committed
9566b47
Preserve dynamic shape expressions across XLA shape reconstruction (#13)
stevenvar
committed
7599afe
Reject dynamic shape expressions in MLIR XLA kernels (#19)
stevenvar
committed
a1af363
Preserve dynamic shape expressions through constant folding (#22)
stevenvar
committed
9ba6e97
Prevent CPU HLO profiling crashes (#30)
stevenvar
committed
d7b764e
Avoid MLIR-only lowering with dynamic sizes (#31)
stevenvar
committed
c7d4072
Preserve dynamic expressions through binary broadcasts (#38)
stevenvar
committed
4751e85
Fix dynamic expression propagation gaps found by tests (#39)
stevenvar
committed
2f52a26
Propagate symbolic Size expressions through Range (#41)

stevenvar
and
fenxcc
committed
f894ecb
Port dynamic input expression clustering stack (#40)
stevenvar
committed
b57854d
Propagate symbolic Size values through TensorFlow inference (#42)
stevenvar
committed
7b3187c
Share dynamic shape expressions between TensorFlow and XLA (#35)
stevenvar
committed
f53f061
Handle singleton and inconsistent dynamic solve candidates (#28)
stevenvar
committed
b072b26
Preserve shape-derived expressions through value chains (#50)
stevenvar
committed
e36fb78

Commits on Aug 17, 2026

Add core XLA dynamic expression invariant tests (#52)
stevenvar
committed
635bc63

Commits on Aug 18, 2026

Fix dynamic expression edge cases and batch bucketing
stevenvar
committed
d6a8e51