Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dcuda_device_functions.h114 __device__ inline unsigned CudaLaneId() { in CudaLaneId() function
134 unsigned src_dst_mask = 1u << CudaLaneId() | 1u << src_lane; in CudaValidateShuffleSyncMask()
145 int lane_id = CudaLaneId(); in CudaShuffleGetSrcLane()
153 unsigned lane_id = CudaLaneId(); in CudaShuffleUpGetSrcLane()
163 unsigned lane_id = CudaLaneId(); in CudaShuffleDownGetSrcLane()
172 int lane_id = CudaLaneId(); in CudaShuffleXorGetSrcLane()
192 assert(mask & 1u << CudaLaneId());
201 assert(mask & 1u << CudaLaneId()); in CudaBallotSync()
212 assert(mask & 1u << CudaLaneId()); in CudaAnySync()
223 assert(mask & 1u << CudaLaneId()); in CudaAllSync()
Dcuda_kernel_helper_test.cu.cc93 unsigned lane_id = CudaLaneId(); in CudaShuffleGetSrcLaneTest()