Searched refs:GpuContext (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 42 class GpuContext; variable
71 static port::StatusOr<GpuDeviceHandle> DeviceFromContext(GpuContext* context);
77 static bool CreateStream(GpuContext* context, GpuStreamHandle* stream);
83 static void DestroyStream(GpuContext* context, GpuStreamHandle* stream);
93 static port::Status CreateEvent(GpuContext* context, GpuEventHandle* result,
99 static port::Status DestroyEvent(GpuContext* context, GpuEventHandle* event);
104 static void* DeviceAllocate(GpuContext* context, uint64 bytes);
109 static void DeviceDeallocate(GpuContext* context, void* location);
115 static void* UnifiedMemoryAllocate(GpuContext* context, uint64 bytes);
121 static void UnifiedMemoryDeallocate(GpuContext* context, void* location);
[all …]
|
| D | gpu_timer.cc | 28 GpuContext* context = parent_->gpu_context(); in Init()
52 GpuContext* context = parent_->gpu_context(); in Destroy()
|
| D | gpu_activation.cc | 25 GpuContext* ExtractGpuContext(GpuExecutor* gpu_exec);
|
| D | gpu_executor.h | 236 GpuContext* gpu_context();
323 GpuContext* context_;
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_driver.h | 27 class GpuContext {
29 GpuContext(CUcontext context, int64 id) : context_(context), id_(id) {} in GpuContext() function
35 GpuContext(GpuContext&&) = delete;
36 GpuContext(const GpuContext&) = delete;
37 GpuContext& operator=(GpuContext&&) = delete;
38 GpuContext& operator=(const GpuContext&) = delete;
55 using CudaContext = gpu::GpuContext;
|
| D | cuda_driver.cc | 72 static GpuContext* Add(CUcontext context) { in Add()
79 it->second = MakeUnique<GpuContext>(context, next_id_++); in Add()
95 static std::map<CUcontext, std::unique_ptr<GpuContext>>* Live() { in Live()
97 new std::map<CUcontext, std::unique_ptr<GpuContext>>; in Live()
198 GpuContext* context; // Only valid if id == a known good context.
206 ScopedActivateContext::ScopedActivateContext(GpuContext* cuda_context) { in ScopedActivateContext()
395 GpuContext** context) { in CreateContext()
470 /* static */ void GpuDriver::DestroyContext(GpuContext* context) { in DestroyContext()
514 GpuDriver::ContextGetSharedMemConfig(GpuContext* context) { in ContextGetSharedMemConfig()
532 GpuContext* context, CUsharedconfig shared_mem_config) { in ContextSetSharedMemConfig()
[all …]
|
| D | cuda_gpu_executor.cc | 113 GpuContext* ExtractGpuContext(GpuExecutor* cuda_exec) { in ExtractGpuContext()
961 GpuContext* GpuExecutor::gpu_context() { return context_; } in gpu_context()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 54 class GpuContext { class
56 GpuContext(const int v) : device_ordinal_(v) {} in GpuContext() function in stream_executor::gpu::GpuContext
61 GpuContext(GpuContext&&) = delete;
62 GpuContext(const GpuContext&) = delete;
63 GpuContext& operator=(GpuContext&&) = delete;
64 GpuContext& operator=(const GpuContext&) = delete;
174 ScopedActivateContext::ScopedActivateContext(GpuContext* context) { in ScopedActivateContext()
283 GpuContext fromCtx(from_pointerAttributes.device); in ROCMPointersToCanAccessString()
284 GpuContext toCtx(to_pointerAttributes.device); in ROCMPointersToCanAccessString()
364 GpuContext** context) { in CreateContext()
[all …]
|
| D | rocm_gpu_executor.cc | 90 static GpuContext* GetGpuContext(Stream* stream) { in GetGpuContext()
95 GpuContext* ExtractGpuContext(GpuExecutor* rocm_exec) { in ExtractGpuContext()
863 GpuContext* GpuExecutor::gpu_context() { return context_; } in gpu_context()
|