xla

ac10c398 - Adjust to xla_client -> computation_client change