xla

9d217b11 - Increase the default XLA_IR_SHAPE_CACHE_SIZE to accomunate the larger model size (#4937)