Searched defs:address_base (Results 1 – 2 of 2) sorted by relevance
189 auto address_base = threadIdx.x + threadIdx.y * blockDim.x; in welford_merge_block_vertical() local974 int address_base = m_offset * stride + c_offset; in batch_norm_collect_statistics_channels_last_kernel() local1117 int address_base = m_offset * stride + c_offset; in batch_norm_transform_input_channels_last_kernel() local1142 auto address_base = threadIdx.x + threadIdx.y * blockDim.x; in merge_block_vertical_backward() local1204 int address_base = m_offset * stride + c_offset; in batch_norm_backward_reduce_channels_last_kernel() local1355 int address_base = m_offset * stride + c_offset; in batch_norm_backward_elemt_channels_last_kernel_impl() local
642 int address_base = threadIdx.x + threadIdx.y*blockDim.x; in block_x_reduce() local