Home
last modified time | relevance | path

Searched refs:CudaGridRangeX (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsegment_reduction_ops_gpu.cu.cc60 for (int stripe_index : CudaGridRangeX(total_stripe_count)) { in SortedSegmentSumCustomKernel()
115 for (int input_index : CudaGridRangeX(input_total_size)) { in UnsortedSegmentCustomKernel()
/external/tensorflow/tensorflow/core/util/
Dcuda_device_functions.h89 __device__ detail::CudaGridRange<T> CudaGridRangeX(T count) { in CudaGridRangeX() function
388 for (int i : CudaGridRangeX(count)) { in SetZero()
399 for (int i : CudaGridRangeX(count)) { in SetToValue()
Dcuda_kernel_helper.h28 for (int i : ::tensorflow::CudaGridRangeX<int>(n))