Searched refs:gpuStream_t (Results 1 – 9 of 9) 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()
|
| D | cuda_sparse.h | 37 using gpuStream_t = cudaStream_t; variable
56 using gpuStream_t = hipStream_t; variable
453 gpuStream_t gpu_stream_;
|
| D | gpu_device_functions.h | 48 using gpuStream_t = cudaStream_t; variable
61 using gpuStream_t = hipStream_t; variable
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | where_op_gpu.cu.h | 71 gpuStream_t stream = 0, in operator()
86 gpuStream_t stream = 0,
104 gpuStream_t stream = 0,
123 gpuStream_t stream = 0,
|
| D | histogram_op_gpu.cu.cc | 69 const gpuStream_t& stream = GetGpuStream(context); in Compute()
|
| D | reduction_gpu_kernels.cu.h | 651 const gpuStream_t& cu_stream) {
719 const gpuStream_t& cu_stream) {
763 const gpuStream_t& cu_stream) {
811 T init, const gpuStream_t& cu_stream) {
854 const gpuStream_t& cu_stream) {
874 T init, const gpuStream_t& cu_stream) {
889 const gpuStream_t& cu_stream) {
951 const gpuStream_t& cu_stream) {
1077 const gpuStream_t& cu_stream = GetGpuStream(ctx);
|
| D | bincount_op_gpu.cu.cc | 59 const gpuStream_t& stream = GetGpuStream(context); in Compute()
|
| D | topk_op_gpu.h | 382 cudaError LaunchTopKKernel(const gpuStream_t& stream, int num_shards,
|
| /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()
765 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize()
1038 const gpuStream_t* gpu_stream = reinterpret_cast<const gpuStream_t*>( in ReinitializeDevice()
|