Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/distributed_runtime/rpc/
Dgrpc_session_test.cc633 const string& dev_a = cluster->devices()[0].name(); in TEST() local
635 LOG(INFO) << "master " << master << "dev_a " << dev_a << "dev_b " << dev_b; in TEST()
646 a->set_assigned_device_name(dev_a); in TEST()
648 a_err->set_assigned_device_name(dev_a); in TEST()
650 a2->set_assigned_device_name(dev_a); in TEST()
690 const string& dev_a = cluster->devices()[0].name(); in TEST() local
692 LOG(INFO) << "master " << master << "dev_a " << dev_a << "dev_b " << dev_b; in TEST()
703 a->set_assigned_device_name(dev_a); in TEST()
705 a_err->set_assigned_device_name(dev_a); in TEST()
707 a2->set_assigned_device_name(dev_a); in TEST()
[all …]
/external/tensorflow/tensorflow/core/util/
Dcuda_solvers.cc547 int m, int n, int k, const Scalar* dev_a, in UnmqrImpl() argument
554 bufsize(cusolver_dn_handle, side, trans, m, n, k, CUDAComplex(dev_a), lda, in UnmqrImpl()
561 cusolver_dn_handle, side, trans, m, n, k, CUDAComplex(dev_a), lda, in UnmqrImpl()
573 int m, int n, int k, const Scalar* dev_a, int lda, \
579 dev_a, lda, dev_tau, dev_c, ldc, dev_lapack_info); \
592 int n, int k, Scalar* dev_a, int lda, in UngqrImpl() argument
598 CUDAComplex(dev_a), lda, in UngqrImpl()
605 solver(cusolver_dn_handle, m, n, k, CUDAComplex(dev_a), lda, in UngqrImpl()
613 Status CudaSolver::Ungqr(int m, int n, int k, Scalar* dev_a, int lda, \
617 context_, cusolver_dn_handle_, m, n, k, dev_a, lda, \
Dcuda_solvers.h305 int k, const Scalar* dev_a, int lda, const Scalar* dev_tau,
315 Status Ungqr(int m, int n, int k, Scalar* dev_a, int lda,