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

Open

stevenvar wants to merge 15 commits into

steven.for-serving-2.20-clean-historyfrom

steven.fix-clean-history-review-findings

Commits on Aug 18, 2026

Add shared dynamic shape expression representation
stevenvar
committed
04f6d85
Propagate symbolic shapes and values through TensorFlow inference
stevenvar
committed
c83b134
Preserve dynamic expressions through HLO construction and inference
stevenvar
committed
eded2c7
Propagate dynamic expressions through tf2xla kernels
stevenvar
committed
15f1c18
Preserve dynamic expressions through binary broadcasting
stevenvar
committed
797aa93
Preserve shape-derived expressions through constant folding
stevenvar
committed
1b6dcea
Cluster graphs using normalized dynamic subexpressions
stevenvar
committed
a962070
Compile and execute clusters with dynamic expressions
stevenvar
committed
3d1a28c
Guard MLIR lowering for dynamic shape expressions
stevenvar
committed
9a26c5d
Pass dynamic batch dimensions through CPU execution
stevenvar
committed
676bfdf
Generate runtime-dependent CPU loop bounds
stevenvar
committed
3ac507f
Support dynamic dimensions in CPU operation emitters
stevenvar
committed
6eb28c5
Prevent CPU HLO profiling crashes
stevenvar
committed
ce28e7a
Add dynamic expression integration tests
stevenvar
committed
c3d2540
Fix dynamic expression edge cases and batch bucketing
stevenvar
committed
85612da