Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc447 return DoBlasInternal(wrap::rocblas_sasum, stream, in DoBlasAsum()
455 return DoBlasInternal(wrap::rocblas_dasum, stream, in DoBlasAsum()
463 return DoBlasInternal(wrap::rocblas_scasum, stream, in DoBlasAsum()
471 return DoBlasInternal(wrap::rocblas_dzasum, stream, in DoBlasAsum()
480 return DoBlasInternal(wrap::rocblas_saxpy, stream, in DoBlasAxpy()
489 return DoBlasInternal(wrap::rocblas_daxpy, stream, in DoBlasAxpy()
498 return DoBlasInternal( in DoBlasAxpy()
507 return DoBlasInternal( in DoBlasAxpy()
515 return DoBlasInternal(wrap::rocblas_scopy, stream, in DoBlasCopy()
523 return DoBlasInternal(wrap::rocblas_dcopy, stream, in DoBlasCopy()
[all …]
Drocm_blas.h111 bool DoBlasInternal(FuncT rocblas_func, Stream *stream,
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc580 return DoBlasInternal(cublasSasum, stream, false /* = pointer_mode_host */, in DoBlasInternalImpl()
588 return DoBlasInternal(cublasDasum, stream, false /* = pointer_mode_host */, in DoBlasInternalImpl()
596 return DoBlasInternal(cublasScasum, stream, false /* = pointer_mode_host */, in DoBlasInternalImpl()
604 return DoBlasInternal(cublasDzasum, stream, false /* = pointer_mode_host */, in DoBlasInternalImpl()
612 return DoBlasInternal(cublasSaxpy, stream, true /* = pointer_mode_host */, in DoBlasInternalImpl()
620 return DoBlasInternal(cublasDaxpy, stream, true /* = pointer_mode_host */, in DoBlasInternalImpl()
630 return DoBlasInternal(cublasCaxpy, stream, true /* = pointer_mode_host */, in DoBlasInternalImpl()
641 return DoBlasInternal(cublasZaxpy, stream, true /* = pointer_mode_host */, in DoBlasInternalImpl()
650 return DoBlasInternal(cublasScopy, stream, true /* = pointer_mode_host */, in DoBlasInternalImpl()
658 return DoBlasInternal(cublasDcopy, stream, true /* = pointer_mode_host */, in DoBlasInternalImpl()
[all …]
Dcuda_blas.h97 bool DoBlasInternal(FuncT cublas_func, Stream *stream, bool pointer_mode_host, in DoBlasInternal() function