Searched refs:cublas_func (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.h | 84 bool DoBlasInternalImpl(FuncT cublas_func, Stream *stream,
91 bool DoBlasInternal(FuncT cublas_func, Stream *stream, bool pointer_mode_host, in DoBlasInternal() argument
93 return DoBlasInternalImpl(cublas_func, stream, pointer_mode_host, in DoBlasInternal()
98 bool DoBlasInternalFailureOK(FuncT cublas_func, Stream *stream, in DoBlasInternalFailureOK() argument
102 return DoBlasInternalImpl(cublas_func, stream, pointer_mode_host, in DoBlasInternalFailureOK()
111 FuncT cublas_func, Stream *stream, blas::Transpose transa,
|
| D | cuda_blas.cc | 402 bool CUDABlas::DoBlasInternalImpl(FuncT cublas_func, Stream *stream, in DoBlasInternalImpl() argument
426 cublasStatus_t ret = cublas_func(blas_, args...); in DoBlasInternalImpl()
2145 FuncT cublas_func, Stream *stream, blas::Transpose transa, in DoBlasGemmBatchedInternal() argument
2245 cublas_func, stream, true /* = pointer_mode_host */, in DoBlasGemmBatchedInternal()
|