Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu_impl.cu.cc104 IntType local_col = gidx - curr_offset; in concat_variable_kernel() local
111 input_ptr[gidy * segment_width + local_col]; in concat_variable_kernel()
Dsplit_lib_gpu.cu.cc159 IntType local_col = gidx - curr_offset; in split_v_kernel() local
165 output_ptr[gidy * segment_width + local_col] = in split_v_kernel()