Searched refs:gpu_stream (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | custom_call_thunk.cc | 55 auto gpu_stream = se::gpu::AsGpuStreamValue(params.stream); in ExecuteOnStream() local
56 using call_type = void (*)(decltype(gpu_stream), void** /*buffers*/, in ExecuteOnStream()
59 typed_call_target(gpu_stream, buffers.data(), opaque_.data(), opaque_.size()); in ExecuteOnStream()
|
| D | BUILD | 651 "//tensorflow/stream_executor/gpu:gpu_stream",
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | BUILD | 97 ":gpu_stream",
154 hdrs = if_gpu_is_configured(["gpu_stream.h"]),
163 name = "gpu_stream",
164 srcs = if_gpu_is_configured(["gpu_stream.cc"]),
165 hdrs = if_gpu_is_configured(["gpu_stream.h"]),
194 ":gpu_stream",
|
| D | gpu_stream.h | 70 GpuStreamHandle gpu_stream() const { in gpu_stream() function
76 GpuStreamHandle cuda_stream() const { return gpu_stream(); } in cuda_stream()
|
| D | gpu_timer.cc | 76 parent_->gpu_context(), start_event_, stream->gpu_stream()); in Start()
85 parent_->gpu_context(), stop_event_, stream->gpu_stream()); in Stop()
|
| D | gpu_event.cc | 41 stream->gpu_stream()); in Record()
|
| D | gpu_stream.cc | 58 return AsGpuStream(stream)->gpu_stream(); in AsGpuStreamValue()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_device.cc | 122 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize() argument
133 stream_ = gpu_stream; in Reinitialize()
765 void Reinitialize(OpKernelContext* context, const gpuStream_t* gpu_stream, in Reinitialize() argument
768 stream_device_.Reinitialize(context, gpu_stream, tf_gpu_id, base_allocator, in Reinitialize()
1038 const gpuStream_t* gpu_stream = reinterpret_cast<const gpuStream_t*>( in ReinitializeDevice() local
1040 concrete_device->Reinitialize(context, gpu_stream, tf_gpu_id_, allocator, in ReinitializeDevice()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | BUILD | 112 "//tensorflow/stream_executor/gpu:gpu_stream",
|
| D | rocm_gpu_executor.cc | 592 if (GpuDriver::WaitStreamOnEvent(context_, AsGpuStream(stream)->gpu_stream(), in WaitForEvent()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | BUILD | 581 "//tensorflow/stream_executor/gpu:gpu_stream",
|
| D | cuda_gpu_executor.cc | 684 if (GpuDriver::WaitStreamOnEvent(context_, AsGpuStream(stream)->gpu_stream(), in WaitForEvent()
|