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.cc3532 const auto &cuda_algo = in DoBlasInternalImpl() local
3568 if (cuda_algo.workspace_size()) { in DoBlasInternalImpl()
3570 &workspace, scratch_allocator, cuda_algo.workspace_size()); in DoBlasInternalImpl()
3575 << cuda_algo.algo_id() << " requiring " in DoBlasInternalImpl()
3576 << cuda_algo.workspace_size() << " bytes of workspace"; in DoBlasInternalImpl()
3637 cuda_plan.d_desc(), cuda_algo.algo(), workspace, in DoBlasInternalImpl()
3638 cuda_algo.workspace_size(), cuda_stream); in DoBlasInternalImpl()