Home
last modified time | relevance | path

Searched defs:blas_lt (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_blas.h69 BlasLt &blas_lt() { return blas_lt_; } in blas_lt() function
H A Dcuda_blas_lt.cc124 cublasLtHandle_t blas_lt; in Init() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dmatmul_utils.cc695 se::cuda::BlasLt* blas_lt = se::cuda::GetBlasLt(stream); in DoMatmul() local
760 se::cuda::BlasLt* blas_lt = se::cuda::GetBlasLt(stream); in GetAlgorithms() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dmatmul_util.cc122 se::cuda::BlasLt* blas_lt = se::cuda::GetBlasLt(stream); in GetPlanAndAlgorithms() local