Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dcuda_device_functions.h171 __device__ inline unsigned CudaShuffleXorGetSrcLane(int lane_mask, int width) { in CudaShuffleXorGetSrcLane() argument
173 int src_lane = lane_id ^ lane_mask; in CudaShuffleXorGetSrcLane()
320 __device__ T CudaShuffleXorSync(unsigned mask, T value, int lane_mask,
324 mask, detail::CudaShuffleXorGetSrcLane(lane_mask, width)));
326 return __shfl_xor_sync(mask, value, lane_mask, width);
328 return __shfl_xor(value, lane_mask, width);
336 int lane_mask,
340 hi = CudaShuffleXorSync(mask, hi, lane_mask, width);
341 lo = CudaShuffleXorSync(mask, lo, lane_mask, width);
Dcuda_kernel_helper.h90 unsigned mask, Eigen::half value, int lane_mask, int width = warpSize) {
92 CudaShuffleXorSync(mask, static_cast<uint16>(value), lane_mask, width));
/external/u-boot/drivers/pci/
Dpci_mvebu.c76 u32 lane_mask; member