Searched refs:kCudaWarpAll (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_kernel_helper_test.cu.cc | 106 CudaShuffleSync(kCudaWarpAll, lane_id, src_lane, width); in CudaShuffleGetSrcLaneTest()
112 CudaShuffleUpSync(kCudaWarpAll, lane_id, delta, width); in CudaShuffleGetSrcLaneTest()
118 CudaShuffleDownSync(kCudaWarpAll, lane_id, delta, width); in CudaShuffleGetSrcLaneTest()
124 CudaShuffleXorSync(kCudaWarpAll, lane_id, lane_lane, width); in CudaShuffleGetSrcLaneTest()
|
| D | cuda_device_functions.h | 111 const unsigned kCudaWarpAll = 0xffffffff; variable
191 __device__ inline void CudaSyncWarp(unsigned mask = kCudaWarpAll) {
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | bias_op_gpu.cu.cc | 188 data += CudaShuffleXorSync(kCudaWarpAll, data, delta); in BiasGradNCHW_SharedAtomics()
|
| D | depthwise_conv_op_gpu.h | 1267 unsigned active_threads = CudaBallotSync(kCudaWarpAll, channel_in_range);
1534 unsigned active_threads = CudaBallotSync(kCudaWarpAll, channel_in_range);
|