Searched refs:gpuStream_t (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | gpu_kernel_helper.h | 48 using gpuStream_t = cudaStream_t; variable
52 using gpuStream_t = hipStream_t; variable
86 inline const gpuStream_t& GetGpuStream(OpKernelContext* context) { in GetGpuStream()
87 const gpuStream_t* ptr = CHECK_NOTNULL( in GetGpuStream()
88 reinterpret_cast<const gpuStream_t*>(context->op_device_context() in GetGpuStream()
101 size_t shared_memory_size_bytes, gpuStream_t stream, in GpuLaunchKernel()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | where_op_gpu.cu.h | 84 gpuStream_t stream = 0, in operator()
99 gpuStream_t stream = 0,
117 gpuStream_t stream = 0,
136 gpuStream_t stream = 0,
|
| D | cuda_sparse.h | 36 using gpuStream_t = cudaStream_t; variable
47 using gpuStream_t = hipStream_t; variable
388 gpuStream_t gpu_stream_;
|
| D | bincount_op_gpu.cu.cc | 68 const gpuStream_t& stream = GetGpuStream(context); in Compute()
|
| D | histogram_op_gpu.cu.cc | 79 const gpuStream_t& stream = GetGpuStream(context); in Compute()
|
| D | reduction_gpu_kernels.cu.h | 655 const gpuStream_t& cu_stream) {
723 const gpuStream_t& cu_stream) {
767 const gpuStream_t& cu_stream) {
815 T init, const gpuStream_t& cu_stream) {
858 const gpuStream_t& cu_stream) {
878 T init, const gpuStream_t& cu_stream) {
893 const gpuStream_t& cu_stream) {
955 const gpuStream_t& cu_stream) {
1077 const gpuStream_t& cu_stream = GetGpuStream(ctx);
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 89 typedef cudaStream_t gpuStream_t; typedef
96 typedef hipStream_t gpuStream_t;
122 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize()
140 const gpuStream_t& stream() const override { return *stream_; } in stream()
202 static void CUDART_CB asyncFree(gpuStream_t stream, cudaError_t status, in asyncFree()
205 static void asyncFree(gpuStream_t stream, hipError_t status, void* userData) { in asyncFree()
218 const gpuStream_t* stream_; // Not owned. in asyncFree()
708 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize()
895 const gpuStream_t* gpu_stream = reinterpret_cast<const gpuStream_t*>( in ReinitializeDevice()
|