Use tensorflow/compiler/xla/stream_executor instead of tensorflow/stream_executor
Use tensorflow/compiler/xla/stream_executor instead of tensorflow/stream_executor

PiperOrigin-RevId: 470804752

google/jax,google/jax,google/jax,google/jax