Searched refs:ldainv (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cuda_solvers.cc | 786 int ldainv, DeviceLapackInfo* dev_lapack_info, int batch_size) { in GetriBatchedImpl() argument
805 ldainv, dev_lapack_info->mutable_data(), batch_size)); in GetriBatchedImpl()
814 int ldainv, DeviceLapackInfo* dev_lapack_info, int batch_size) { \
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
846 reinterpret_cast<CudaScalar**>(dev_a_inv_dev_ptrs.mutable_data()), ldainv, in MatInvBatchedImpl()
855 const Scalar* const host_a_inv_dev_ptrs[], int ldainv, \
861 ldainv, dev_lapack_info, batch_size); \
|
| D | cuda_solvers.h | 255 const Scalar* const host_a_inverse_dev_ptrs[], int ldainv,
265 int ldainv, DeviceLapackInfo* dev_lapack_info,
|