| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| gpu_stream.cc | 26 if (!GpuDriver::CreateStream(parent_->gpu_context(), &gpu_stream_)) {
29 return GpuDriver::CreateEvent(parent_->gpu_context(), &completed_event_,
37 GpuDriver::DestroyEvent(parent_->gpu_context(), &completed_event_);
43 GpuDriver::DestroyStream(parent_->gpu_context(), &gpu_stream_);
47 return GpuDriver::IsStreamIdle(parent_->gpu_context(), gpu_stream_);
|
| gpu_event.cc | 31 return GpuDriver::CreateEvent(parent_->gpu_context(), &gpu_event_,
36 return GpuDriver::DestroyEvent(parent_->gpu_context(), &gpu_event_);
40 return GpuDriver::RecordEvent(parent_->gpu_context(), gpu_event_,
|
| gpu_timer.cc | 28 GpuContext* context = parent_->gpu_context();
52 GpuContext* context = parent_->gpu_context();
70 parent_->gpu_context(), &elapsed_milliseconds, start_event_, stop_event_);
76 parent_->gpu_context(), start_event_, stream->gpu_stream());
85 parent_->gpu_context(), stop_event_, stream->gpu_stream());
|
| gpu_executor.h | 236 GpuContext* gpu_context();
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| cuda_event.cc | 27 GpuDriver::QueryEvent(parent_->gpu_context(), gpu_event_);
|
| cuda_gpu_executor.cc | 115 return cuda_exec->gpu_context();
961 GpuContext* GpuExecutor::gpu_context() { return context_; } function in class:stream_executor::gpu::GpuExecutor
[all...] |
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| rocm_event.cc | 26 GpuDriver::QueryEvent(parent_->gpu_context(), gpu_event_);
|
| rocm_gpu_executor.cc | 92 ->gpu_context();
97 return rocm_exec->gpu_context();
863 GpuContext* GpuExecutor::gpu_context() { return context_; } function in class:stream_executor::gpu::GpuExecutor
[all...] |
| /external/libdrm/amdgpu/ |
| amdgpu_cs.c | 54 struct amdgpu_context *gpu_context; local
62 gpu_context = calloc(1, sizeof(struct amdgpu_context));
63 if (!gpu_context)
66 gpu_context->dev = dev;
68 r = pthread_mutex_init(&gpu_context->sequence_mutex, NULL);
81 gpu_context->id = args.out.alloc.ctx_id;
85 list_inithead(&gpu_context->sem_list[i][j][k]);
86 *context = (amdgpu_context_handle)gpu_context;
91 pthread_mutex_destroy(&gpu_context->sequence_mutex);
92 free(gpu_context);
[all...] |
