Searched refs:GetGpuStream (Results 1 – 12 of 12) sorted by relevance
69 const gpuStream_t& stream = GetGpuStream(context); in Compute()
140 const auto& cu_stream = GetGpuStream(ctx);267 const auto& cu_stream = GetGpuStream(ctx);
339 const auto& cu_stream = GetGpuStream(c); in RadixSort()372 const auto& cu_stream = GetGpuStream(c); in CountAndSortParts()
456 const auto& cu_stream = GetGpuStream(ctx);570 const auto& cu_stream = GetGpuStream(context);
59 const gpuStream_t& stream = GetGpuStream(context); in Compute()
213 const auto& cu_stream = GetGpuStream(context); in Compute()
1077 const gpuStream_t& cu_stream = GetGpuStream(ctx);
86 inline const gpuStream_t& GetGpuStream(OpKernelContext* context) { in GetGpuStream() function
240 const auto& cu_stream = GetGpuStream(ctx); in LSTMBlockCellFpropWithCUDA()381 const auto& cu_stream = GetGpuStream(ctx); in LSTMBlockCellBpropWithCUDA()
381 auto cuda_stream = tensorflow::GetGpuStream(context); in CountIf()414 auto cuda_stream = GetGpuStream(context); in DoNMS()
346 auto cuda_stream = GetGpuStream(context); in Compute()
56 const auto& cu_stream = GetGpuStream(c); in operator ()()