Searched defs:cublas_func (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.h | 91 bool DoBlasInternal(FuncT cublas_func, Stream *stream, bool pointer_mode_host, in DoBlasInternal()
98 bool DoBlasInternalFailureOK(FuncT cublas_func, Stream *stream, in DoBlasInternalFailureOK()
|
| D | cuda_blas.cc | 402 bool CUDABlas::DoBlasInternalImpl(FuncT cublas_func, Stream *stream, in DoBlasInternalImpl()
2145 FuncT cublas_func, Stream *stream, blas::Transpose transa, in DoBlasGemmBatchedInternal()
|