Searched defs:dev_pivots (Results 1 – 2 of 2) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | cuda_solvers.cc | 450 int n, Scalar* A, int lda, int* dev_pivots, in GetrfImpl()
768 int lda, int* dev_pivots, in GetrfBatchedImpl()
803 const Scalar* const host_a_dev_ptrs[], int lda, const int* dev_pivots, in GetrsBatchedImpl()
850 int lda, const int* dev_pivots, const Scalar* const host_a_inv_dev_ptrs[], in GetriBatchedImpl()
|
| H A D | rocm_solvers.cc | 772 int* dev_pivots, in MatInvBatchedImpl()
|