xla_compiler_.CompileFunction(options, function, args, result));
}

Status TfGraphToHloCompiler::CompileSingleOp(
    const XlaCompiler::CompileOptions& options, const OpKernelContext* ctx,
    absl::Span<const XlaArgument> args, XlaCompilationResult* result) {
  return ADD_SOURCE_LOCATION(xla_compiler_.CompileSingleOp(
      options, XlaCompiler::SingleOpCompileArgument(*ctx), args, result));
}

                 XlaCompilationResult* result) override;

  // Compiles a Tensorflow single op into an HloModuleProto stored in the
  // XlaCompilationResult pointed to by `result` by calling
  // XlaCompiler::CompileSingleOp.
  Status CompileSingleOp(const XlaCompiler::CompileOptions& options,
                         const OpKernelContext* ctx,
                         absl::Span<const XlaArgument> args,

                         XlaCompilationResult* result) = 0;

  // Compiles a Tensorflow single op to an HloModuleProto stored in the
  // XlaCompilationResult pointed to by `result`.
  virtual Status CompileSingleOp(const XlaCompiler::CompileOptions& options,
                                 const OpKernelContext* ctx,
                                 absl::Span<const XlaArgument> args,

  DeviceCompilerClient<ExecutableType, ClientType>* compiler_client() {
    return compiler_client_.get();
  }

  string DebugString() const override;

 private:
  // Common implementation of Compile and CompileSingleOp. The `OpKernelContext`
  // parameter is always null for the former.
  Status CompileImpl(
      const XlaCompiler::CompileOptions& compile_options,
      const XlaCompiler::Options& options, const NameAttrList& function,