Searched refs:curr_segment (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dconcat_lib_gpu_impl.cu.cc95 IntType curr_segment = segment; local
98 while ((curr_col_offset = col_scan[curr_segment + 1]) <= gidx) {
100 ++curr_segment;
105 const T* input_ptr = input_ptrs[curr_segment];
H A Dsplit_lib_gpu.cu.cc145 IntType curr_segment = segment; local
148 while ((curr_col_offset = col_scan[curr_segment + 1]) <= gidx) {
150 ++curr_segment;
155 T* output_ptr = output_ptrs[curr_segment];

Completed in 61 milliseconds