Searched defs:thrx (Results 1 – 1 of 1) sorted by relevance
166 const int thrx = threadIdx.x + threadIdx.y * blockDim.x; in compute_stats() local243 const int thrx = threadIdx.x + threadIdx.y * blockDim.x; in vectorized_layer_norm_kernel_impl() local1047 const int thrx = threadIdx.x + threadIdx.y * blockDim.x; in cuComputeGradInput() local