Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dgpu_kernel_helper_test.cu.cc117 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()
Dgpu_device_functions.h193 __device__ const unsigned kCudaWarpAll = 0xffffffff; variable
298 __device__ inline void GpuSyncWarp(unsigned mask = kCudaWarpAll) {
/external/tensorflow/tensorflow/core/kernels/
Dbias_op_gpu.cu.cc191 data += GpuShuffleXorSync(kCudaWarpAll, data, delta); in BiasGradNCHW_SharedAtomics()
Ddepthwise_conv_op_gpu.h1263 unsigned active_threads = GpuBallotSync(kCudaWarpAll, channel_in_range);
1531 unsigned active_threads = GpuBallotSync(kCudaWarpAll, channel_in_range);