Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.cc548 int lda, const Scalar* dev_tau, Scalar* dev_c, in UnmqrImpl() argument
555 CUDAComplex(dev_tau), CUDAComplex(dev_c), ldc, &lwork)); in UnmqrImpl()
562 CUDAComplex(dev_tau), CUDAComplex(dev_c), ldc, in UnmqrImpl()
574 const Scalar* dev_tau, Scalar* dev_c, int ldc, \
579 dev_a, lda, dev_tau, dev_c, ldc, dev_lapack_info); \
593 const Scalar* dev_tau, int* dev_lapack_info) { in UngqrImpl() argument
599 CUDAComplex(dev_tau), &lwork)); in UngqrImpl()
606 CUDAComplex(dev_tau), CUDAComplex(dev_workspace.mutable_data()), in UngqrImpl()
614 const Scalar* dev_tau, int* dev_lapack_info) { \
618 dev_tau, dev_lapack_info); \
Dcuda_solvers.h292 Status Geqrf(int m, int n, Scalar* dev_A, int lda, Scalar* dev_tau,
305 int k, const Scalar* dev_a, int lda, const Scalar* dev_tau,
316 const Scalar* dev_tau, int* dev_lapack_info) TF_MUST_USE_RESULT;