Searched refs:host_a_inv_dev_ptrs (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.cc | 785 int lda, const int* dev_pivots, const Scalar* const host_a_inv_dev_ptrs[], in GetriBatchedImpl() argument
797 host_a_inv_dev_ptrs, dev_a_inv_dev_ptrs.bytes())) { in GetriBatchedImpl()
813 const int* dev_pivots, const Scalar* const host_a_inv_dev_ptrs[], \
819 host_a_inv_dev_ptrs, ldainv, dev_lapack_info, batch_size); \
828 int lda, const Scalar* const host_a_inv_dev_ptrs[], int ldainv, in MatInvBatchedImpl() argument
840 host_a_inv_dev_ptrs, dev_a_inv_dev_ptrs.bytes())) { in MatInvBatchedImpl()
855 const Scalar* const host_a_inv_dev_ptrs[], int ldainv, \
860 host_a_dev_ptrs, lda, host_a_inv_dev_ptrs, \
|