Home
last modified time | relevance | path

Searched refs:curr_col_offset (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu_impl.cu.cc98 IntType curr_col_offset; in concat_variable_kernel() local
99 while ((curr_col_offset = col_scan[curr_segment + 1]) <= gidx) { in concat_variable_kernel()
100 curr_offset = curr_col_offset; in concat_variable_kernel()
105 IntType segment_width = curr_col_offset - curr_offset; in concat_variable_kernel()
Dsplit_lib_gpu.cu.cc153 IntType curr_col_offset; in split_v_kernel() local
154 while ((curr_col_offset = col_scan[curr_segment + 1]) <= gidx) { in split_v_kernel()
155 curr_offset = curr_col_offset; in split_v_kernel()
160 IntType segment_width = curr_col_offset - curr_offset; in split_v_kernel()