Home
last modified time | relevance | path

Searched refs:dev_A (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.h219 Status Potrf(cublasFillMode_t uplo, int n, Scalar* dev_A, int lda,
238 Status Getrf(int m, int n, Scalar* dev_A, int lda, int* dev_pivots,
292 Status Geqrf(int m, int n, Scalar* dev_A, int lda, Scalar* dev_tau,
322 Scalar* dev_A, int lda,
331 Status Gesvd(signed char jobu, signed char jobvt, int m, int n, Scalar* dev_A,
335 Status GesvdjBatched(cusolverEigMode_t jobz, int m, int n, Scalar* dev_A,
Dcuda_solvers.cc632 int n, Scalar* dev_A, int lda, in HeevdImpl() argument
639 CUDAComplex(dev_A), lda, in HeevdImpl()
646 solver(cusolver_dn_handle, jobz, uplo, n, CUDAComplex(dev_A), lda, in HeevdImpl()
655 int n, Scalar* dev_A, int lda, \
660 context_, cusolver_dn_handle_, jobz, uplo, n, dev_A, lda, \
693 signed char jobu, signed char jobvt, int m, int n, Scalar* dev_A, \
698 cusolver_dn_handle_, jobu, jobvt, m, n, dev_A, lda, \
736 cusolverEigMode_t jobz, int m, int n, Scalar* dev_A, int lda, \
741 context_, cusolver_dn_handle_, jobz, m, n, dev_A, \
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcusolver_context.h58 Status Potrf(se::blas::UpperLower uplo, int n, se::DeviceMemory<T> dev_A,
103 Status Potrf(se::blas::UpperLower uplo, int n, se::DeviceMemory<T> dev_A,