Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_blas.h69 BlasLt &blas_lt() { return blas_lt_; } in blas_lt() function
Dcuda_blas_lt.cc124 cublasLtHandle_t blas_lt; in Init() local
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_util.cc122 se::cuda::BlasLt* blas_lt = se::cuda::GetBlasLt(stream); in GetPlanAndAlgorithms() local
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
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