Method set_arity

tensorflow/stream_executor/gpu/gpu_kernel.h:49–49

As arity cannot be reflected upon using the CUDA API, the arity is explicitly set during the GpuExecutor::GetKernel initialization process.

  // As arity cannot be reflected upon using the CUDA API, the arity is
  // explicitly set during the GpuExecutor::GetKernel initialization process.
  void set_arity(unsigned arity) { arity_ = arity; }
  unsigned Arity() const override { return arity_; }

  // Returns the GpuFunctionHandle value for passing to the CUDA API.