Searched refs:DoBlasInternalImpl (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.h | 81 bool DoBlasInternalImpl(FuncT rocblas_func, Stream *stream,
90 return DoBlasInternalImpl(rocblas_func, stream, pointer_mode_host, in DoBlasInternal()
96 return DoBlasInternalImpl(rocblas_func, stream, pointer_mode_host, in DoBlasInternalFailureOK()
|
| D | rocm_blas.cc | 382 bool ROCMBlas::DoBlasInternalImpl(FuncT rocblas_func, Stream *stream, in DoBlasInternalImpl() function in stream_executor::gpu::ROCMBlas
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.h | 84 bool DoBlasInternalImpl(FuncT cublas_func, Stream *stream,
93 return DoBlasInternalImpl(cublas_func, stream, pointer_mode_host, in DoBlasInternal()
102 return DoBlasInternalImpl(cublas_func, stream, pointer_mode_host, in DoBlasInternalFailureOK()
|
| D | cuda_blas.cc | 402 bool CUDABlas::DoBlasInternalImpl(FuncT cublas_func, Stream *stream, in DoBlasInternalImpl() function in stream_executor::gpu::CUDABlas
1608 return DoBlasInternalImpl( in DoBlasGemm()
2229 ok = DoBlasInternalImpl( in DoBlasGemmBatchedInternal()
2377 bool ok = DoBlasInternalImpl( in DoBlasGemmStridedBatched()
2401 bool ok = DoBlasInternalImpl( in DoBlasGemmStridedBatched()
|