Searched refs:max_workspace_size (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_blas.h | 156 size_t max_workspace_size,
|
| D | cuda_blas.cc | 3462 size_t max_workspace_size, in DoBlasInternalImpl() argument
3467 CreateCublasLtMatmulPreference(plan, max_workspace_size)); in DoBlasInternalImpl()
3518 size_t max_workspace_size, in DoBlasInternalImpl() argument
3520 return GetBlasLtMatmulAlgorithmsInternal(plan, max_workspace_size, in DoBlasInternalImpl()
3589 size_t max_workspace_size = 4 * 1024 * 1024; // 4 MiB in DoBlasInternalImpl() local
3591 GetBlasLtMatmulAlgorithmsInternal(plan, max_workspace_size, in DoBlasInternalImpl()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream_executor_pimpl.cc | 353 size_t max_workspace_size, in GetBlasLtMatmulAlgorithms() argument
360 return blas_support->GetBlasLtMatmulAlgorithms(plan, max_workspace_size, in GetBlasLtMatmulAlgorithms()
|
| D | stream_executor_pimpl.h | 411 size_t max_workspace_size, int max_algorithm_count);
|
| D | blas.h | 1467 size_t max_workspace_size,
2336 size_t max_workspace_size, \
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_blas.cc | 2665 size_t max_workspace_size, in GetBlasLtMatmulAlgorithms() argument
|