Searched refs:src_lane (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_device_functions.h | 133 unsigned src_lane) { in CudaValidateShuffleSyncMask() argument
134 unsigned src_dst_mask = 1u << CudaLaneId() | 1u << src_lane; in CudaValidateShuffleSyncMask()
136 unsigned src_lane_mask = __shfl_sync(mask, mask, src_lane); in CudaValidateShuffleSyncMask()
138 unsigned src_lane_mask = __shfl(mask, src_lane); in CudaValidateShuffleSyncMask()
144 __device__ inline unsigned CudaShuffleGetSrcLane(int src_lane, int width) { in CudaShuffleGetSrcLane() argument
147 int lane_offset = src_lane & width - 1; in CudaShuffleGetSrcLane()
173 int src_lane = lane_id ^ lane_mask; in CudaShuffleXorGetSrcLane() local
174 if (src_lane > (lane_id | width - 1)) { in CudaShuffleXorGetSrcLane()
177 return src_lane; in CudaShuffleXorGetSrcLane()
234 __device__ T CudaShuffleSync(unsigned mask, T value, int src_lane,
[all …]
|
| D | cuda_kernel_helper_test.cu.cc | 103 for (int src_lane = -warpSize; src_lane <= warpSize; ++src_lane) { in CudaShuffleGetSrcLaneTest() local
104 unsigned actual_lane = detail::CudaShuffleGetSrcLane(src_lane, width); in CudaShuffleGetSrcLaneTest()
106 CudaShuffleSync(kCudaWarpAll, lane_id, src_lane, width); in CudaShuffleGetSrcLaneTest()
107 check_result("Shuffle", src_lane, actual_lane, expect_lane); in CudaShuffleGetSrcLaneTest()
|
| D | cuda_kernel_helper.h | 71 int src_lane,
74 CudaShuffleSync(mask, static_cast<uint16>(value), src_lane, width));
|
| /external/v8/src/arm/ |
| D | macro-assembler-arm.h | 438 void ReplaceLane(QwNeonRegister dst, QwNeonRegister src, Register src_lane,
441 SwVfpRegister src_lane, int lane);
|
| D | macro-assembler-arm.cc | 1012 Register src_lane, NeonDataType dt, int lane) { in ReplaceLane() argument
1021 vmov(dt, double_dst, double_lane, src_lane); in ReplaceLane()
1025 SwVfpRegister src_lane, int lane) { in ReplaceLane() argument
1028 VmovExtended(s_code, src_lane.code()); in ReplaceLane()
|
| /external/v8/src/arm64/ |
| D | simulator-logic-arm64.cc | 1401 uint64_t src_lane = src.Uint(vform, i); in sli() local
1403 uint64_t shifted = src_lane << shift; in sli()
1441 uint64_t src_lane = src.Uint(vform, i); in sri() local
1449 shifted = src_lane >> shift; in sri()
|
| /external/vixl/src/aarch64/ |
| D | logic-aarch64.cc | 1532 uint64_t src_lane = src.Uint(vform, i); in sli() local
1534 uint64_t shifted = src_lane << shift; in sli()
1584 uint64_t src_lane = src.Uint(vform, i); in sri() local
1592 shifted = src_lane >> shift; in sri()
|