xla

205e3071 - move partition outside of extract compile graph