Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcublas_lt_matmul_thunk.cc48 plan_.GetAlgorithms(params.stream)); in ExecuteOnStream()
H A Dgpu_conv_algorithm_picker.cc109 StatusOr<std::vector<MaybeFusedConvRunner>> GetAlgorithms( in GetAlgorithms() function
712 GetAlgorithms(config, stream, cudnn_frontend_enabled, in PickBestAlgorithmNoCacheCuda()
733 GetAlgorithms(config, stream, cudnn_frontend_enabled, in PickBestAlgorithmNoCacheCuda()
H A Dmatmul_utils.h142 StatusOr<std::vector<se::cuda::BlasLt::MatmulAlgorithm>> GetAlgorithms(
H A Dgemm_algorithm_picker.cc287 plan.GetAlgorithms(stream)); in DoGemmAutotune()
H A Dmatmul_utils.cc759 MatmulPlan::GetAlgorithms(se::Stream* stream) const { in GetAlgorithms() function in xla::gpu::cublas_lt::MatmulPlan
H A Djitrt_custom_calls.cc609 auto algos = plan->GetAlgorithms(stream); in operator ()()