message: Opt out DEVICE_GPU_XLA_JIT and DEVICE_XLA_GPU from ResizeNearestNeighborOp, ResizeBilinearOp, and ResizeBilinearGradOp, because the dilation-based approach may introduce convolutions too large for GPU to handle.
diff: @@ -587,7 +587,13 @@ void ResizeNearestNeighborOp::Compile(XlaOpKernelContext* ctx) {
  GeneralCompile(ctx, align_corners_, is_kernel_bilinear_);
  }
  -REGISTER_XLA_OP(Name("ResizeNearestNeighbor").CompileTimeConstantInput("size"),
  +REGISTER_XLA_OP(Name("ResizeNearestNeighbor")
  + .Device(DEVICE_CPU_XLA_JIT)
  + .CompileTimeConstantInput("size"),
  + ResizeNearestNeighborOp);
  +REGISTER_XLA_OP(Name("ResizeNearestNeighbor")
  + .Device(DEVICE_XLA_CPU)
  + .CompileTimeConstantInput("size"),
  ResizeNearestNeighborOp);
  ResizeBilinearOp::ResizeBilinearOp(OpKernelConstruction* ctx)
  @@ -604,7 +610,13 @@ void ResizeBilinearOp::Compile(XlaOpKernelContext* ctx) {
  GeneralCompile(ctx, align_corners_, is_kernel_bilinear_);
  }
  -REGISTER_XLA_OP(Name("ResizeBilinear").CompileTimeConstantInput("size"),
  +REGISTER_XLA_OP(Name("ResizeBilinear")
  + .Device(DEVICE_CPU_XLA_JIT)
  + .CompileTimeConstantInput("size"),
  + ResizeBilinearOp);
  +REGISTER_XLA_OP(Name("ResizeBilinear")
  + .Device(DEVICE_XLA_CPU)
  + .CompileTimeConstantInput("size"),
  ResizeBilinearOp);
  ResizeBilinearGradOp::ResizeBilinearGradOp(OpKernelConstruction* ctx)
  @@ -698,6 +710,7 @@ void ResizeBilinearGradOp::Compile(XlaOpKernelContext* ctx) {
  ctx->SetOutput(0, output);
  }
  -REGISTER_XLA_OP(Name("ResizeBilinearGrad"), ResizeBilinearGradOp);
  +REGISTER_XLA_OP(Name("ResizeBilinearGrad").Device(DEVICE_CPU_XLA_JIT), ResizeBilinearGradOp);
  +REGISTER_XLA_OP(Name("ResizeBilinearGrad").Device(DEVICE_XLA_CPU), ResizeBilinearGradOp);
  } // namespace tensorflow
  