Home
last modified time | relevance | path

Searched refs:blas_preferred_backend (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DContext.cpp268 if (blas_preferred_backend == at::BlasBackend::Cublaslt) { in blasPreferredBackend()
281 if (hipblaslt_unsupported) blas_preferred_backend = at::BlasBackend::Cublas; in blasPreferredBackend()
284 return blas_preferred_backend; in blasPreferredBackend()
303 blas_preferred_backend = b; in setBlasPreferredBackend()
H A DContext.h387 at::BlasBackend blas_preferred_backend = variable