Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_blas_lt.h132 struct MatmulAlgorithm { struct
149 port::StatusOr<std::vector<MatmulAlgorithm>> GetMatmulAlgorithms( argument
159 const MatmulAlgorithm& algorithm,
204 const MatmulAlgorithm& algorithm,
H A Dcuda_blas_lt.cc226 port::StatusOr<std::vector<BlasLt::MatmulAlgorithm>>
247 std::vector<BlasLt::MatmulAlgorithm> algorithms; in GetMatmulAlgorithms()
261 const BlasLt::MatmulAlgorithm& algorithm, in DoMatmul()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dmatmul_utils.h138 const se::cuda::BlasLt::MatmulAlgorithm& algorithm,
142 StatusOr<std::vector<se::cuda::BlasLt::MatmulAlgorithm>> GetAlgorithms(
158 const se::cuda::BlasLt::MatmulAlgorithm& algorithm,
H A Dgemm_algorithm_picker.cc286 std::vector<se::cuda::BlasLt::MatmulAlgorithm> algorithms, in DoGemmAutotune()
291 GetBestAlgorithm<se::cuda::BlasLt::MatmulAlgorithm>( in DoGemmAutotune()
294 [&](const se::cuda::BlasLt::MatmulAlgorithm& algorithm) in DoGemmAutotune()
H A Dcublas_lt_matmul_thunk.h50 std::optional<se::cuda::BlasLt::MatmulAlgorithm> algorithm_;
H A Dcublas_lt_matmul_thunk.cc47 std::vector<se::cuda::BlasLt::MatmulAlgorithm> algorithms, in ExecuteOnStream()
H A Dmatmul_utils.cc692 const se::cuda::BlasLt::MatmulAlgorithm& algorithm, in DoMatmul()
721 const se::cuda::BlasLt::MatmulAlgorithm& algorithm, in ExecuteOnStream()
758 StatusOr<std::vector<se::cuda::BlasLt::MatmulAlgorithm>>
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmatmul_util.h62 std::vector<se::cuda::BlasLt::MatmulAlgorithm> algorithms;
91 const se::cuda::BlasLt::MatmulAlgorithm& algorithm,
H A Dmatmul_op_fused.cc185 const std::vector<se::cuda::BlasLt::MatmulAlgorithm>& algorithms, in AutotuneMatmul()
303 const se::cuda::BlasLt::MatmulAlgorithm& algorithm, in operator ()()
309 se::cuda::BlasLt::MatmulAlgorithm algorithm = algorithms[0]; in operator ()()
H A Dmatmul_util.cc187 std::vector<se::cuda::BlasLt::MatmulAlgorithm> algorithms, in GetPlanAndAlgorithms()