xla

0efa9e46 - Add `sym_sizes` with dynamic shape support (#3909)

Add `sym_sizes` with dynamic shape support (#3909) * add sym_sizes with dynamic shape support * test fixes * remove an unnecessary cast * fix test failures * addderss Jack's feedback Co-authored-by: Nikolay Korovaiko <korovaikon@gmail.com>