message: Use new Xla execution Op
diff: @@ -18,7 +18,7 @@ limitations under the License.
  #include "tensorflow/compiler/jit/xla_device.h"
  #include "tensorflow/compiler/jit/xla_device_ops.h"
  -#include "tensorflow/compiler/jit/kernels/xla_device_launch_op.h"
  +#include "tensorflow/compiler/jit/kernels/xla_launch_op.h"
  #include "tensorflow/compiler/tf2xla/xla_op_registry.h"
  #include "tensorflow/compiler/tf2xla/kernels/gather_op.h"
  @@ -64,7 +64,7 @@ static bool OpFilter(KernelDef* kdef) {
  return true;
  }
  -REGISTER_XLA_LAUNCH_KERNEL(DEVICE_XLA_IPU, XlaDeviceLaunchOp, kIpuAllTypes);
  +REGISTER_XLA_LAUNCH_KERNEL(DEVICE_XLA_IPU, XlaLocalLaunchOp, kIpuAllTypes);
  REGISTER_XLA_DEVICE_KERNELS(DEVICE_XLA_IPU, kIpuAllTypes);
  REGISTER_XLA_BACKEND(DEVICE_IPU_XLA_JIT, kIpuAllTypes, OpFilter);
  