Home
last modified time | relevance | path

Searched defs:gpu_executor (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Drocm_solvers.cc121 GpuExecutor* gpu_executor = static_cast<GpuExecutor*>( in GpuSolver() local
732 static inline Status TrsmImpl(GpuExecutor* gpu_executor, SolverFnT solver, in TrsmImpl()
769 Status MatInvBatchedImpl(GpuExecutor* gpu_executor, SolverFnT solver, in MatInvBatchedImpl()
809 Status GeamImpl(GpuExecutor* gpu_executor, SolverFnT solver, in GeamImpl()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Doutfeed_manager.cc34 stream_executor::gpu::GpuExecutor *gpu_executor = in GetOrCreateOutfeedManager() local
H A Dinfeed_manager.cc92 stream_executor::gpu::GpuExecutor* gpu_executor = in GetOrCreateInfeedManager() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_dnn.cc4590 Stream* stream, CudnnHandle& cudnn, GpuExecutor* gpu_executor, in CreateOpRunners()