HomeSort by relevance Sort by last modified time
    Searched refs:col_scan (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
concat_lib_gpu_impl.cu.cc 65 IntType* col_scan = GetCudaDeviceArrayOnDevice(&output_scan); local
80 smem_col_scan[i] = col_scan[i];
85 col_scan = smem_col_scan;
92 cuda_helper::upper_bound<IntType>(col_scan, num_inputs, gidx) - 1;
94 IntType curr_offset = col_scan[segment];
98 while ((curr_col_offset = col_scan[curr_segment + 1]) <= gidx) {
split_lib_gpu.cu.cc 115 IntType* col_scan = GetCudaDeviceArrayOnDevice(&output_scan); local
130 smem_col_scan[i] = col_scan[i];
135 col_scan = smem_col_scan;
142 cuda_helper::upper_bound<IntType>(col_scan, num_outputs, gidx) - 1;
144 IntType curr_offset = col_scan[segment];
148 while ((curr_col_offset = col_scan[curr_segment + 1]) <= gidx) {

Completed in 121 milliseconds