Searched refs:pointer_mode_host (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.h | 82 bool pointer_mode_host, bool err_on_failure,
89 bool pointer_mode_host, Args... args) { in DoBlasInternal() argument
90 return DoBlasInternalImpl(rocblas_func, stream, pointer_mode_host, in DoBlasInternal()
95 bool pointer_mode_host, Args... args) { in DoBlasInternalFailureOK() argument
96 return DoBlasInternalImpl(rocblas_func, stream, pointer_mode_host, in DoBlasInternalFailureOK()
|
| D | rocm_blas.cc | 383 bool pointer_mode_host, bool err_on_failure, in DoBlasInternalImpl() argument
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.h | 85 bool pointer_mode_host, bool err_on_failure,
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()
99 bool pointer_mode_host, Args... args) { in DoBlasInternalFailureOK() argument
102 return DoBlasInternalImpl(cublas_func, stream, pointer_mode_host, in DoBlasInternalFailureOK()
|
| D | cuda_blas.cc | 403 bool pointer_mode_host, bool err_on_failure, in DoBlasInternalImpl() argument
414 if (!pointer_mode.Init(pointer_mode_host ? CUBLAS_POINTER_MODE_HOST in DoBlasInternalImpl()
|