| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | gpu_driver.h | 39 class GpuContext; variable
68 static port::StatusOr<GpuDeviceHandle> DeviceFromContext(GpuContext* context);
74 static bool CreateStream(GpuContext* context, GpuStreamHandle* stream,
81 static void DestroyStream(GpuContext* context, GpuStreamHandle* stream);
91 static port::Status InitEvent(GpuContext* context, GpuEventHandle* result,
97 static port::Status DestroyEvent(GpuContext* context, GpuEventHandle* event);
102 static void* DeviceAllocate(GpuContext* context, uint64 bytes);
107 static void DeviceDeallocate(GpuContext* context, void* location);
113 static void* UnifiedMemoryAllocate(GpuContext* context, uint64 bytes);
119 static void UnifiedMemoryDeallocate(GpuContext* context, void* location);
[all …]
|
| D | asm_compiler.h | 28 class GpuContext; variable
81 gpu::GpuContext* context, std::vector<CubinOrPTXImage> images);
|
| 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 | 234 GpuContext* gpu_context();
322 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 | 92 static GpuContext* Add(CUcontext context) { in Add()
99 it->second = absl::make_unique<GpuContext>(context, next_id_++); in Add()
115 static std::map<CUcontext, std::unique_ptr<GpuContext>>* Live() { in Live()
117 new std::map<CUcontext, std::unique_ptr<GpuContext>>; in Live()
191 GpuContext* context; // Only valid if id == a known good context.
199 ScopedActivateContext::ScopedActivateContext(GpuContext* cuda_context) { in ScopedActivateContext()
392 GpuContext** context) { in CreateContext()
468 /* static */ void GpuDriver::DestroyContext(GpuContext* context) { in DestroyContext()
502 GpuDriver::ContextGetSharedMemConfig(GpuContext* context) { in ContextGetSharedMemConfig()
511 GpuContext* context, CUsharedconfig shared_mem_config) { in ContextSetSharedMemConfig()
[all …]
|
| D | cuda_asm_compiler.cc | 36 gpu::GpuContext* context, std::vector<CubinOrPTXImage> images) { in LinkGpuAsm()
|
| D | cuda_driver_test.cc | 55 GpuContext se_context1(context1, /*id=*/101); in TEST()
|
| D | cuda_gpu_executor.cc | 113 GpuContext* ExtractGpuContext(GpuExecutor* cuda_exec) { in ExtractGpuContext()
892 GpuContext* GpuExecutor::gpu_context() { return context_; } in gpu_context()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_driver.cc | 63 class GpuContext { class
65 GpuContext(const int v) : device_ordinal_(v) {} in GpuContext() function in stream_executor::gpu::GpuContext
70 GpuContext(GpuContext&&) = delete;
71 GpuContext(const GpuContext&) = delete;
72 GpuContext& operator=(GpuContext&&) = delete;
73 GpuContext& operator=(const GpuContext&) = delete;
175 ScopedActivateContext::ScopedActivateContext(GpuContext* context) { in ScopedActivateContext()
284 GpuContext fromCtx(from_pointerAttributes.device); in ROCMPointersToCanAccessString()
285 GpuContext toCtx(to_pointerAttributes.device); in ROCMPointersToCanAccessString()
354 GpuContext** context) { in CreateContext()
[all …]
|
| D | rocm_gpu_executor.cc | 89 static GpuContext* GetGpuContext(Stream* stream) { in GetGpuContext()
94 GpuContext* ExtractGpuContext(GpuExecutor* rocm_exec) { in ExtractGpuContext()
797 GpuContext* GpuExecutor::gpu_context() { return context_; } in gpu_context()
|
| /external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| D | gpu_virtual_mem_allocator_test.cc | 29 using ::stream_executor::gpu::GpuContext;
42 GpuContext* gpu_context = reinterpret_cast<GpuContext*>( in CreateAllocator()
55 GpuContext* gpu_context = reinterpret_cast<GpuContext*>( in TEST()
|
| 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,
81 stream_executor::gpu::GpuContext& gpu_context_;
|
| D | gpu_virtual_mem_allocator.cc | 27 using ::stream_executor::gpu::GpuContext;
64 GpuContext& gpu_context, PlatformGpuId gpu_id, in Create()
113 const std::vector<Visitor>& free_visitors, GpuContext& gpu_context, in GpuVirtualMemAllocator()
|
| D | gpu_process_state.cc | 102 auto* gpu_context = reinterpret_cast<stream_executor::gpu::GpuContext*>( in CreateSubAllocator()
|
| D | gpu_bfc_allocator_test.cc | 69 auto* gpu_context = reinterpret_cast<stream_executor::gpu::GpuContext*>( in CreateVirtualMemorySubAllocator()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 453 return LinkGpuAsm(static_cast<se::gpu::GpuContext*>( in LinkModules()
|