Searched refs:input_copy_ptrs_base (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/linalg/ |
| D | matrix_solve_op.cc | 223 const Scalar** input_copy_ptrs_base = in ComputeAsync() local
226 input_copy_ptrs_base[batch] = &input_copy_reshaped(batch, 0, 0); in ComputeAsync()
232 solver->GetrfBatched(n, input_copy_ptrs_base, n, pivots_mat.data(), in ComputeAsync()
286 const Scalar** input_copy_ptrs_base = in ComputeAsync() local
292 input_copy_ptrs_base[batch] = &input_copy_reshaped(batch, 0, 0); in ComputeAsync()
299 input_copy_ptrs_base, n, pivots_mat.data(), in ComputeAsync()
|
| D | determinant_op.cc | 207 const Scalar** input_copy_ptrs_base = in ComputeAsync() local
210 input_copy_ptrs_base[batch] = &input_copy_reshaped(batch, 0, 0); in ComputeAsync()
216 solver->GetrfBatched(n, input_copy_ptrs_base, n, pivots_mat.data(), in ComputeAsync()
351 const Scalar** input_copy_ptrs_base = in ComputeAsync() local
354 input_copy_ptrs_base[batch] = &input_copy_reshaped(batch, 0, 0); in ComputeAsync()
360 solver->GetrfBatched(n, input_copy_ptrs_base, n, pivots_mat.data(), in ComputeAsync()
|