__global__ void TimeTwoCudaKernel()

in tensorflow_time_two/cc/kernels/time_two_kernels.cu.cc [30:35]


__global__ void TimeTwoCudaKernel(const int size, const T* in, T* out) {
  for (int i = blockIdx.x * blockDim.x + threadIdx.x; i < size;
       i += blockDim.x * gridDim.x) {
    out[i] = 2 * ldg(in + i);
  }
}