Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dcuda_device_functions.h136 unsigned src_lane_mask = __shfl_sync(mask, mask, src_lane); in CudaValidateShuffleSyncMask() local
138 unsigned src_lane_mask = __shfl(mask, src_lane); in CudaValidateShuffleSyncMask() local
140 return (src_dst_mask & ~mask) == 0 && src_lane_mask == mask; in CudaValidateShuffleSyncMask()