| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_stream.cc | 26 if (!GpuDriver::CreateStream(parent_->gpu_context(), &gpu_stream_, in Init()
30 return GpuDriver::InitEvent(parent_->gpu_context(), &completed_event_, in Init()
38 GpuDriver::DestroyEvent(parent_->gpu_context(), &completed_event_); in Destroy()
44 GpuDriver::DestroyStream(parent_->gpu_context(), &gpu_stream_); in Destroy()
48 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::InitEvent(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 | 234 GpuContext* gpu_context();
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_virtual_mem_allocator_test.cc | 42 GpuContext* gpu_context = reinterpret_cast<GpuContext*>( in CreateAllocator() local
45 {}, {}, *gpu_context, gpu_id, in CreateAllocator()
55 GpuContext* gpu_context = reinterpret_cast<GpuContext*>( in TEST() local
58 {}, {}, *gpu_context, gpu_id, in TEST()
67 void* host_mem[2] = {GpuDriver::HostAllocate(gpu_context, kBufSize), in TEST()
68 GpuDriver::HostAllocate(gpu_context, kBufSize)}; in TEST()
73 ASSERT_TRUE(GpuDriver::SynchronousMemcpyH2D(gpu_context, gpu_buf, host_mem[0], in TEST()
76 ASSERT_TRUE(GpuDriver::SynchronousMemcpyD2H(gpu_context, host_mem[1], gpu_buf, in TEST()
|
| D | gpu_virtual_mem_allocator.cc | 64 GpuContext& gpu_context, PlatformGpuId gpu_id, in Create() argument
102 &gpu_context, AlignUp(virtual_address_space_size, max_granularity))); in Create()
107 alloc_visitors, free_visitors, gpu_context, gpu_id, in Create()
113 const std::vector<Visitor>& free_visitors, GpuContext& gpu_context, in GpuVirtualMemAllocator() argument
117 gpu_context_(gpu_context), in GpuVirtualMemAllocator()
|
| D | gpu_virtual_mem_allocator.h | 47 stream_executor::gpu::GpuContext& gpu_context, PlatformGpuId gpu_id,
77 stream_executor::gpu::GpuContext& gpu_context, PlatformGpuId gpu_id,
|
| D | gpu_process_state.cc | 102 auto* gpu_context = reinterpret_cast<stream_executor::gpu::GpuContext*>( in CreateSubAllocator()
121 alloc_visitors, {}, *gpu_context, platform_gpu_id, in CreateSubAllocator()
|
| D | gpu_bfc_allocator_test.cc | 69 auto* gpu_context = reinterpret_cast<stream_executor::gpu::GpuContext*>( in CreateVirtualMemorySubAllocator() local
71 return GpuVirtualMemAllocator::Create({}, {}, *gpu_context, gpu_id, in CreateVirtualMemorySubAllocator()
|
| /external/libdrm/amdgpu/ |
| D | amdgpu_cs.c | 55 struct amdgpu_context *gpu_context; in amdgpu_cs_ctx_create2() local
63 gpu_context = calloc(1, sizeof(struct amdgpu_context)); in amdgpu_cs_ctx_create2()
64 if (!gpu_context) in amdgpu_cs_ctx_create2()
67 gpu_context->dev = dev; in amdgpu_cs_ctx_create2()
69 r = pthread_mutex_init(&gpu_context->sequence_mutex, NULL); in amdgpu_cs_ctx_create2()
82 gpu_context->id = args.out.alloc.ctx_id; in amdgpu_cs_ctx_create2()
86 list_inithead(&gpu_context->sem_list[i][j][k]); in amdgpu_cs_ctx_create2()
87 *context = (amdgpu_context_handle)gpu_context; in amdgpu_cs_ctx_create2()
92 pthread_mutex_destroy(&gpu_context->sequence_mutex); in amdgpu_cs_ctx_create2()
93 free(gpu_context); in amdgpu_cs_ctx_create2()
|
| /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 | 91 ->gpu_context(); in GetGpuContext()
96 return rocm_exec->gpu_context(); in ExtractGpuContext()
797 GpuContext* GpuExecutor::gpu_context() { return context_; } in gpu_context() function in stream_executor::gpu::GpuExecutor
|
| /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()
892 GpuContext* GpuExecutor::gpu_context() { return context_; } in gpu_context() function in stream_executor::gpu::GpuExecutor
|