Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_gpu_impl.cu.cc74 IntType* smem_col_scan = reinterpret_cast<IntType*>(smem); in concat_variable_kernel() local
81 smem_col_scan[i] = col_scan[i]; in concat_variable_kernel()
86 col_scan = smem_col_scan; in concat_variable_kernel()
Dsplit_lib_gpu.cu.cc129 IntType* smem_col_scan = reinterpret_cast<IntType*>(smem); in split_v_kernel() local
136 smem_col_scan[i] = col_scan[i]; in split_v_kernel()
141 col_scan = smem_col_scan; in split_v_kernel()