Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc3484 const auto &cuda_algo = in DoBlasInternalImpl() local
3520 if (cuda_algo.workspace_size()) { in DoBlasInternalImpl()
3522 &workspace, scratch_allocator, cuda_algo.workspace_size()); in DoBlasInternalImpl()
3527 << cuda_algo.algo_id() << " requiring " in DoBlasInternalImpl()
3528 << cuda_algo.workspace_size() << " bytes of workspace"; in DoBlasInternalImpl()
3589 cuda_plan.d_desc(), cuda_algo.algo(), workspace, in DoBlasInternalImpl()
3590 cuda_algo.workspace_size(), cuda_stream); in DoBlasInternalImpl()