xla

0cf761e8 - Increase the default XLA_IR_SHAPE_CACHE_SIZE to accomunate the larger model size