xla

1492eba8 - use the eager_mode_context for some of the tests (#7276)