Searched refs:ldainv (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | cuda_solvers.cc | 844 int ldainv, DeviceLapackInfo* dev_lapack_info, int batch_size) { in GetriBatchedImpl() argument
863 ldainv, dev_lapack_info->mutable_data(), batch_size)); in GetriBatchedImpl()
872 int ldainv, DeviceLapackInfo* dev_lapack_info, int batch_size) { \
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
904 reinterpret_cast<CudaScalar**>(dev_a_inv_dev_ptrs.mutable_data()), ldainv, in MatInvBatchedImpl()
913 const Scalar* const host_a_inv_dev_ptrs[], int ldainv, \
919 ldainv, dev_lapack_info, batch_size); \
|
| D | cuda_solvers.h | 274 const Scalar* const host_a_inverse_dev_ptrs[], int ldainv,
284 int ldainv, DeviceLapackInfo* dev_lapack_info,
|