Searched refs:ExtractGpuContext (Results 1 – 3 of 3) sorted by relevance
25 GpuContext* ExtractGpuContext(GpuExecutor* gpu_exec);31 new ScopedActivateContext{ExtractGpuContext(gpu_exec)}) {} in ScopedActivateExecutorContext()
94 GpuContext* ExtractGpuContext(GpuExecutor* rocm_exec) { in ExtractGpuContext() function
113 GpuContext* ExtractGpuContext(GpuExecutor* cuda_exec) { in ExtractGpuContext() function