HomeSort by relevance Sort by last modified time
    Searched refs:CudaShuffleSync (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/util/
cuda_kernel_helper.h 68 __device__ inline Eigen::half CudaShuffleSync(unsigned mask, Eigen::half value,
72 CudaShuffleSync(mask, static_cast<uint16>(value), src_lane, width));
cuda_device_functions.h 234 __device__ T CudaShuffleSync(unsigned mask, T value, int src_lane,
249 __device__ inline double CudaShuffleSync(unsigned mask, double value,
253 hi = CudaShuffleSync(mask, hi, src_lane, width);
254 lo = CudaShuffleSync(mask, lo, src_lane, width);
cuda_kernel_helper_test.cu.cc 104 CudaShuffleSync(kCudaWarpAll, lane_id, src_lane, width);

Completed in 105 milliseconds