Home
last modified time | relevance | path

Searched refs:DoBlasInternal (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc436 return DoBlasInternal(cublasSasum, stream, false /* = pointer_mode_host */, in DoBlasAsum()
444 return DoBlasInternal(cublasDasum, stream, false /* = pointer_mode_host */, in DoBlasAsum()
452 return DoBlasInternal(cublasScasum, stream, false /* = pointer_mode_host */, in DoBlasAsum()
460 return DoBlasInternal(cublasDzasum, stream, false /* = pointer_mode_host */, in DoBlasAsum()
468 return DoBlasInternal(cublasSaxpy, stream, true /* = pointer_mode_host */, in DoBlasAxpy()
476 return DoBlasInternal(cublasDaxpy, stream, true /* = pointer_mode_host */, in DoBlasAxpy()
485 return DoBlasInternal(cublasCaxpy, stream, true /* = pointer_mode_host */, in DoBlasAxpy()
495 return DoBlasInternal(cublasZaxpy, stream, true /* = pointer_mode_host */, in DoBlasAxpy()
504 return DoBlasInternal(cublasScopy, stream, true /* = pointer_mode_host */, in DoBlasCopy()
512 return DoBlasInternal(cublasDcopy, stream, true /* = pointer_mode_host */, in DoBlasCopy()
[all …]
Dcuda_blas.h91 bool DoBlasInternal(FuncT cublas_func, Stream *stream, bool pointer_mode_host, in DoBlasInternal() function
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc403 return DoBlasInternal(wrap::rocblas_sasum, stream, in DoBlasAsum()
411 return DoBlasInternal(wrap::rocblas_dasum, stream, in DoBlasAsum()
435 return DoBlasInternal(wrap::rocblas_saxpy, stream, in DoBlasAxpy()
443 return DoBlasInternal(wrap::rocblas_daxpy, stream, in DoBlasAxpy()
469 return DoBlasInternal(wrap::rocblas_scopy, stream, in DoBlasCopy()
477 return DoBlasInternal(wrap::rocblas_dcopy, stream, in DoBlasCopy()
502 return DoBlasInternal( in DoBlasDot()
511 return DoBlasInternal( in DoBlasDot()
555 return DoBlasInternal(wrap::rocblas_snrm2, stream, in DoBlasNrm2()
563 return DoBlasInternal(wrap::rocblas_dnrm2, stream, in DoBlasNrm2()
[all …]
Drocm_blas.h88 bool DoBlasInternal(FuncT rocblas_func, Stream *stream, in DoBlasInternal() function