xla

c6f09b41 - attempt to return compiled graph instead