Searched refs:CudaGridRange (Results 1 – 2 of 2) sorted by relevance
42 class CudaGridRange {70 __device__ CudaGridRange(T begin, T delta, T end) in CudaGridRange() function89 __device__ detail::CudaGridRange<T> CudaGridRangeX(T count) { in CudaGridRangeX()90 return detail::CudaGridRange<T>(blockIdx.x * blockDim.x + threadIdx.x, in CudaGridRangeX()97 __device__ detail::CudaGridRange<T> CudaGridRangeY(T count) { in CudaGridRangeY()98 return detail::CudaGridRange<T>(blockIdx.y * blockDim.y + threadIdx.y, in CudaGridRangeY()105 __device__ detail::CudaGridRange<T> CudaGridRangeZ(T count) { in CudaGridRangeZ()106 return detail::CudaGridRange<T>(blockIdx.z * blockDim.z + threadIdx.z, in CudaGridRangeZ()
31 for (int i : ::tensorflow::CudaGridRange##axis<int>(n))