Home
last modified time | relevance | path

Searched defs:gpu_context (Results 1 – 9 of 9) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_virtual_mem_allocator.cc64 const std::vector<Visitor>& free_visitors, GpuContext& gpu_context, in Create()
113 const std::vector<Visitor>& free_visitors, GpuContext& gpu_context, in GpuVirtualMemAllocator()
H A Dgpu_virtual_mem_allocator_test.cc42 GpuContext* gpu_context = reinterpret_cast<GpuContext*>( in CreateAllocator() local
55 GpuContext* gpu_context = reinterpret_cast<GpuContext*>( in TEST() local
H A Dgpu_bfc_allocator_test.cc70 auto* gpu_context = reinterpret_cast<stream_executor::gpu::GpuContext*>( in CreateVirtualMemorySubAllocator() local
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/gpu/
H A DGpuWorkloadSketch.cpp41 const GpuWorkloadSketch::Context *GpuWorkloadSketch::gpu_context() const in gpu_context() function in arm_compute::experimental::dynamic_fusion::GpuWorkloadSketch
/aosp_15_r20/external/libdrm/amdgpu/
H A Damdgpu_cs.c55 struct amdgpu_context *gpu_context; in amdgpu_cs_ctx_create2() local
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc805 GpuContext* GpuExecutor::gpu_context() { return context_; } in gpu_context() function in stream_executor::gpu::GpuExecutor
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc934 GpuContext* GpuExecutor::gpu_context() { return context_; } in gpu_context() function in stream_executor::gpu::GpuExecutor
/aosp_15_r20/external/tensorflow/tensorflow/core/runtime_fallback/runtime/
H A Druntime_fallback_kernels.cc235 CUcontext gpu_context = in InjectTfGpuResourcesHelper() local
/aosp_15_r20/external/intel-media-driver/media_driver/linux/common/os/
H A Dmos_os_specific.c4676 GpuContext *gpu_context in Mos_Specific_CreateGpuComputeContext() local