Home
last modified time | relevance | path

Searched refs:GetCudaStream (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dsearchsorted_op_gpu.cu.cc67 const cudaStream_t& stream = GetCudaStream(context); in Compute()
87 const cudaStream_t& stream = GetCudaStream(context); in Compute()
Dbincount_op_gpu.cu.cc58 const cudaStream_t& stream = GetCudaStream(context); in Compute()
Dwhere_op_gpu.cu.h142 const cudaStream_t& cu_stream = GetCudaStream(ctx);
269 const cudaStream_t& cu_stream = GetCudaStream(ctx);
Dhistogram_op_gpu.cu.cc69 const cudaStream_t& stream = GetCudaStream(context); in Compute()
Ddynamic_partition_op_gpu.cu.cc329 const cudaStream_t& cu_stream = GetCudaStream(c); in RadixSort()
362 const cudaStream_t& cu_stream = GetCudaStream(c); in CountAndSortParts()
Dtopk_op_gpu.h438 const cudaStream_t& cu_stream = GetCudaStream(ctx);
552 const cudaStream_t& cu_stream = GetCudaStream(context);
Dsoftmax_op_gpu.cu.cc147 const cudaStream_t& cu_stream = GetCudaStream(context); in Compute()
Dreduction_gpu_kernels.cu.h857 const cudaStream_t& cu_stream = GetCudaStream(ctx);
/external/tensorflow/tensorflow/contrib/rnn/kernels/
Dlstm_ops_gpu.cu.cc235 const cudaStream_t& cu_stream = GetCudaStream(ctx); in LSTMBlockCellFpropWithCUDA()
374 const cudaStream_t& cu_stream = GetCudaStream(ctx); in LSTMBlockCellBpropWithCUDA()
/external/tensorflow/tensorflow/core/util/
Dcuda_launch_config.h293 inline const cudaStream_t& GetCudaStream(OpKernelContext* context) { in GetCudaStream() function