Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_blas_lt.h149 port::StatusOr<std::vector<MatmulAlgorithm>> GetMatmulAlgorithms(
H A Dcuda_blas_lt.cc227 BlasLt::GetMatmulAlgorithms(const BlasLt::MatmulPlan& plan, in GetMatmulAlgorithms() function in stream_executor::cuda::BlasLt
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmatmul_util.cc188 blas_lt->GetMatmulAlgorithms(plan, preference, *max_algorithm_count)); in GetPlanAndAlgorithms()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dmatmul_utils.cc765 return blas_lt->GetMatmulAlgorithms(plan_, preference); in GetAlgorithms()