Searched refs:gpu_executor (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | rocm_solvers.cc | 136 GpuExecutor* gpu_executor = static_cast<GpuExecutor*>( in ROCmSolver() local
152 new ROCmSolverHandles(gpu_executor, hip_stream_)); in ROCmSolver()
208 static inline Status TrsmImpl(GpuExecutor* gpu_executor, SolverFnT solver, in TrsmImpl() argument
218 TF_RETURN_IF_ROCBLAS_ERROR(solver(gpu_executor, rocm_blas_handle, side, uplo, in TrsmImpl()
234 GpuExecutor* gpu_executor = static_cast<GpuExecutor*>( \
236 return TrsmImpl(gpu_executor, BLAS_SOLVER_FN(trsm, type_prefix), \
|
| /external/tensorflow/tensorflow/stream_executor/gpu/ |
| D | BUILD | 105 hdrs = if_gpu_is_configured(["gpu_executor.h"]),
|