Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Dstream_executor_pimpl.cc333 blas::BlasSupport *blas_support = AsBlas(); in GetBlasGemmAlgorithms() local
334 if (!blas_support) { in GetBlasGemmAlgorithms()
337 return blas_support->GetBlasGemmAlgorithms(out_algorithms); in GetBlasGemmAlgorithms()
343 blas::BlasSupport *blas_support = AsBlas(); in CreateBlasLtMatmulPlan() local
344 if (!blas_support) { in CreateBlasLtMatmulPlan()
348 return blas_support->CreateBlasLtMatmulPlan(params); in CreateBlasLtMatmulPlan()
355 blas::BlasSupport *blas_support = AsBlas(); in GetBlasLtMatmulAlgorithms() local
356 if (!blas_support) { in GetBlasLtMatmulAlgorithms()
360 return blas_support->GetBlasLtMatmulAlgorithms(plan, max_workspace_size, in GetBlasLtMatmulAlgorithms()