Searched refs:kCudaWarpAll (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/util/ |
D | gpu_kernel_helper_test.cu.cc | 117 GpuShuffleSync(kCudaWarpAll, lane_id, src_lane, width); in GpuShuffleGetSrcLaneTest() 124 GpuShuffleUpSync(kCudaWarpAll, lane_id, delta, width); in GpuShuffleGetSrcLaneTest() 131 GpuShuffleDownSync(kCudaWarpAll, lane_id, delta, width); in GpuShuffleGetSrcLaneTest() 138 GpuShuffleXorSync(kCudaWarpAll, lane_id, lane_lane, width); in GpuShuffleGetSrcLaneTest()
|
D | gpu_device_functions.h | 193 __device__ const unsigned kCudaWarpAll = 0xffffffff; variable 298 __device__ inline void GpuSyncWarp(unsigned mask = kCudaWarpAll) {
|
/external/tensorflow/tensorflow/core/kernels/ |
D | bias_op_gpu.cu.cc | 191 data += GpuShuffleXorSync(kCudaWarpAll, data, delta); in BiasGradNCHW_SharedAtomics()
|
D | depthwise_conv_op_gpu.h | 1263 unsigned active_threads = GpuBallotSync(kCudaWarpAll, channel_in_range); 1531 unsigned active_threads = GpuBallotSync(kCudaWarpAll, channel_in_range);
|