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

  /external/tensorflow/tensorflow/core/kernels/
concat_lib_gpu_impl.cu.cc 95 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];
split_lib_gpu.cu.cc 145 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 113 milliseconds