Home
last modified time | relevance | path

Searched defs:lane_id (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dcuda_device_functions.h115 unsigned int lane_id; in CudaLaneId() local
145 int lane_id = CudaLaneId(); in CudaShuffleGetSrcLane() local
153 unsigned lane_id = CudaLaneId(); in CudaShuffleUpGetSrcLane() local
163 unsigned lane_id = CudaLaneId(); in CudaShuffleDownGetSrcLane() local
172 int lane_id = CudaLaneId(); in CudaShuffleXorGetSrcLane() local
Dcuda_kernel_helper_test.cu.cc93 unsigned lane_id = CudaLaneId(); in CudaShuffleGetSrcLaneTest() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc2793 llvm::Value* lane_id = reduction_info->GetLaneId(); in EmitEpilogueForReduction() local