Searched refs:warp_id_x (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/contrib/resampler/kernels/
H A Dresampler_ops_gpu.cu.cc170 const int warp_id_x = batch_id * warp_batch_stride + sample_id * 2; local
171 const int warp_id_y = warp_id_x + 1;
172 const T x = warp[warp_id_x];
209 atomicAdd(grad_warp + warp_id_x,

Completed in 139 milliseconds