Searched refs:gpu_context (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_stream.cc | 26 if (!GpuDriver::CreateStream(parent_->gpu_context(), &gpu_stream_)) { in Init()
29 return GpuDriver::CreateEvent(parent_->gpu_context(), &completed_event_, in Init()
37 GpuDriver::DestroyEvent(parent_->gpu_context(), &completed_event_); in Destroy()
43 GpuDriver::DestroyStream(parent_->gpu_context(), &gpu_stream_); in Destroy()
47 return GpuDriver::IsStreamIdle(parent_->gpu_context(), gpu_stream_); in IsIdle()
|
| D | gpu_timer.cc | 28 GpuContext* context = parent_->gpu_context(); in Init()
52 GpuContext* context = parent_->gpu_context(); in Destroy()
70 parent_->gpu_context(), &elapsed_milliseconds, start_event_, stop_event_); in GetElapsedMilliseconds()
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 | 31 return GpuDriver::CreateEvent(parent_->gpu_context(), &gpu_event_, in Init()
36 return GpuDriver::DestroyEvent(parent_->gpu_context(), &gpu_event_); in Destroy()
40 return GpuDriver::RecordEvent(parent_->gpu_context(), gpu_event_, in Record()
|
| D | gpu_executor.h | 236 GpuContext* gpu_context();
|
| /external/libdrm/amdgpu/ |
| D | amdgpu_cs.c | 54 struct amdgpu_context *gpu_context; in amdgpu_cs_ctx_create2() local
62 gpu_context = calloc(1, sizeof(struct amdgpu_context)); in amdgpu_cs_ctx_create2()
63 if (!gpu_context) in amdgpu_cs_ctx_create2()
66 gpu_context->dev = dev; in amdgpu_cs_ctx_create2()
68 r = pthread_mutex_init(&gpu_context->sequence_mutex, NULL); in amdgpu_cs_ctx_create2()
81 gpu_context->id = args.out.alloc.ctx_id; in amdgpu_cs_ctx_create2()
85 list_inithead(&gpu_context->sem_list[i][j][k]); in amdgpu_cs_ctx_create2()
86 *context = (amdgpu_context_handle)gpu_context; in amdgpu_cs_ctx_create2()
91 pthread_mutex_destroy(&gpu_context->sequence_mutex); in amdgpu_cs_ctx_create2()
92 free(gpu_context); in amdgpu_cs_ctx_create2()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_event.cc | 27 GpuDriver::QueryEvent(parent_->gpu_context(), gpu_event_); in PollForStatus()
|
| D | cuda_gpu_executor.cc | 115 return cuda_exec->gpu_context(); in ExtractGpuContext()
961 GpuContext* GpuExecutor::gpu_context() { return context_; } in gpu_context() function in stream_executor::gpu::GpuExecutor
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_event.cc | 26 GpuDriver::QueryEvent(parent_->gpu_context(), gpu_event_); in PollForStatus()
|
| D | rocm_gpu_executor.cc | 92 ->gpu_context(); in GetGpuContext()
97 return rocm_exec->gpu_context(); in ExtractGpuContext()
863 GpuContext* GpuExecutor::gpu_context() { return context_; } in gpu_context() function in stream_executor::gpu::GpuExecutor
|