Searched defs:gpu_context (Results 1 – 6 of 6) sorted by relevance
64 GpuContext& gpu_context, PlatformGpuId gpu_id, in Create()113 const std::vector<Visitor>& free_visitors, GpuContext& gpu_context, in GpuVirtualMemAllocator()
42 GpuContext* gpu_context = reinterpret_cast<GpuContext*>( in CreateAllocator() local55 GpuContext* gpu_context = reinterpret_cast<GpuContext*>( in TEST() local
69 auto* gpu_context = reinterpret_cast<stream_executor::gpu::GpuContext*>( in CreateVirtualMemorySubAllocator() local
55 struct amdgpu_context *gpu_context; in amdgpu_cs_ctx_create2() local
892 GpuContext* GpuExecutor::gpu_context() { return context_; } in gpu_context() function in stream_executor::gpu::GpuExecutor
797 GpuContext* GpuExecutor::gpu_context() { return context_; } in gpu_context() function in stream_executor::gpu::GpuExecutor