Searched refs:cublas_func (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.h | 90 bool DoBlasInternalImpl(FuncT cublas_func, Stream *stream,
97 bool DoBlasInternal(FuncT cublas_func, Stream *stream, bool pointer_mode_host, in DoBlasInternal() argument
99 return DoBlasInternalImpl(cublas_func, stream, pointer_mode_host, in DoBlasInternal()
108 FuncT cublas_func, Stream *stream, blas::Transpose transa,
|
| D | cuda_blas.cc | 532 bool CUDABlas::DoBlasInternalImpl(FuncT cublas_func, Stream *stream, in DoBlasInternalImpl() argument
562 cublasStatus_t ret = cublas_func(blas_, args...); in DoBlasInternalImpl()
2365 FuncT cublas_func, Stream *stream, blas::Transpose transa, in DoBlasInternalImpl() argument
2486 cublas_func, stream, true /* = pointer_mode_host */, in DoBlasInternalImpl()
|