Searched refs:gidy (Results 1 – 2 of 2) sorted by relevance
44 IntType gidy = blockIdx.y * blockDim.y + threadIdx.y; in concat_fixed_kernel() local50 for (; gidy < total_rows; gidy += blockDim.y * gridDim.y) { in concat_fixed_kernel()51 output[gidy * total_cols + gidx] = in concat_fixed_kernel()52 input_ptr[gidy * split_size + col_offset]; in concat_fixed_kernel()108 IntType gidy = blockIdx.y * blockDim.y + threadIdx.y; in concat_variable_kernel() local109 for (; gidy < total_rows; gidy += blockDim.y * gridDim.y) in concat_variable_kernel()110 output[gidy * total_cols + gidx] = in concat_variable_kernel()111 input_ptr[gidy * segment_width + local_col]; in concat_variable_kernel()
163 IntType gidy = blockIdx.y * blockDim.y + threadIdx.y; in split_v_kernel() local164 for (; gidy < total_rows; gidy += blockDim.y * gridDim.y) in split_v_kernel()165 output_ptr[gidy * segment_width + local_col] = in split_v_kernel()166 input_ptr[gidy * total_cols + gidx]; in split_v_kernel()