Searched refs:host_a_inv_dev_ptrs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 843 int lda, const int* dev_pivots, const Scalar* const host_a_inv_dev_ptrs[], in GetriBatchedImpl() argument
855 host_a_inv_dev_ptrs, dev_a_inv_dev_ptrs.bytes())) { in GetriBatchedImpl()
871 const int* dev_pivots, const Scalar* const host_a_inv_dev_ptrs[], \
877 host_a_inv_dev_ptrs, ldainv, dev_lapack_info, batch_size); \
886 int lda, const Scalar* const host_a_inv_dev_ptrs[], int ldainv, in MatInvBatchedImpl() argument
898 host_a_inv_dev_ptrs, dev_a_inv_dev_ptrs.bytes())) { in MatInvBatchedImpl()
913 const Scalar* const host_a_inv_dev_ptrs[], int ldainv, \
918 host_a_dev_ptrs, lda, host_a_inv_dev_ptrs, \
|