/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
|