Searched refs:pivots_mat (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_inverse_op.cc | 157 auto pivots_mat = pivots.template matrix<int>(); in ComputeAsync() local
199 n, pivots_mat.data(), in ComputeAsync()
208 pivots_mat.data(), output_ptr_array_base, n, in ComputeAsync()
222 &pivots_mat(batch, 0), &dev_info.back()(batch)), in ComputeAsync()
236 n, &pivots_mat(batch, 0), in ComputeAsync()
|
| D | determinant_op.cc | 192 auto pivots_mat = pivots.template matrix<int>(); in ComputeAsync() local
216 solver->GetrfBatched(n, input_copy_ptrs_base, n, pivots_mat.data(), in ComputeAsync()
227 &pivots_mat(batch, 0), &dev_info.back()(batch)), in ComputeAsync()
240 pivots_mat.data(), output_reshaped, dev_info.back().mutable_data()); in ComputeAsync()
337 auto pivots_mat = pivots.template matrix<int>(); in ComputeAsync() local
360 solver->GetrfBatched(n, input_copy_ptrs_base, n, pivots_mat.data(), in ComputeAsync()
371 &pivots_mat(batch, 0), &dev_info.back()(batch)), in ComputeAsync()
386 functor(d, input_copy_reshaped_const, pivots_mat.data(), sign_reshaped, in ComputeAsync()
|
| D | matrix_solve_op.cc | 209 auto pivots_mat = pivots.template matrix<int>(); in ComputeAsync() local
232 solver->GetrfBatched(n, input_copy_ptrs_base, n, pivots_mat.data(), in ComputeAsync()
243 &pivots_mat(batch, 0), &dev_info.back()(batch)), in ComputeAsync()
299 input_copy_ptrs_base, n, pivots_mat.data(), in ComputeAsync()
316 &pivots_mat(batch, 0), in ComputeAsync()
|
| D | lu_op_gpu.cu.cc | 168 auto pivots_mat = pivots.template matrix<int32>(); in ComputeAsync() local
197 num_rows, pivots_mat.data(), &dev_info.back(), in ComputeAsync()
210 num_rows, &pivots_mat(batch, 0), &dev_info.back()(batch)), in ComputeAsync()
|