Searched refs:err_on_failure (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.h | 91 bool pointer_mode_host, bool err_on_failure,
143 bool DoBlasLtMatmulInternal(Stream *stream, bool err_on_failure,
|
| D | cuda_blas.cc | 533 bool pointer_mode_host, bool err_on_failure, in DoBlasInternalImpl() argument
563 if ((err_on_failure || VLOG_IS_ON(3)) && ret != CUBLAS_STATUS_SUCCESS) { in DoBlasInternalImpl()
3526 Stream *stream, bool err_on_failure, const blas::IBlasLtMatmulPlan *plan, in DoBlasInternalImpl() argument
3572 if (err_on_failure || VLOG_IS_ON(3)) { in DoBlasInternalImpl()
3595 if (err_on_failure || VLOG_IS_ON(3)) { in DoBlasInternalImpl()
3640 if (err_on_failure || VLOG_IS_ON(3)) { in DoBlasInternalImpl()
3659 if (err_on_failure || VLOG_IS_ON(3)) { in DoBlasInternalImpl()
3676 if (err_on_failure || VLOG_IS_ON(3)) { in DoBlasInternalImpl()
3721 bool err_on_failure = timer != nullptr; in DoBlasInternalImpl() local
3722 bool result = DoBlasLtMatmulInternal(stream, err_on_failure, plan, alpha_cast, in DoBlasInternalImpl()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.h | 105 bool pointer_mode_host, bool err_on_failure,
|
| D | rocm_blas.cc | 427 bool pointer_mode_host, bool err_on_failure, in DoBlasInternalImpl() argument
437 if (err_on_failure && ret != rocblas_status_success) { in DoBlasInternalImpl()
|