Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc352 StreamExecutor::GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan *plan, in GetBlasLtMatmulAlgorithms() function in stream_executor::StreamExecutor
360 return blas_support->GetBlasLtMatmulAlgorithms(plan, max_workspace_size, in GetBlasLtMatmulAlgorithms()
Dstream_executor_pimpl.h410 GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan *plan,
Dblas.h1466 GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan *plan,
2335 GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan *plan, \
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_blas.cc2664 ROCMBlas::GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan *plan, in GetBlasLtMatmulAlgorithms() function in stream_executor::gpu::ROCMBlas
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_blas.cc3517 CUDABlas::GetBlasLtMatmulAlgorithms(const blas::IBlasLtMatmulPlan *plan, in DoBlasInternalImpl() function in stream_executor::gpu::CUDABlas