Searched refs:lu_factor (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | determinant_op_gpu.cu.cc | 87 const Scalar* lu_factor, in DeterminantFromPivotedLUKernel() argument
105 const RealScalar abs_i = Eigen::numext::abs(lu_factor[i_idx]); in DeterminantFromPivotedLUKernel()
107 prod_sign = prod_sign * (lu_factor[i_idx] / abs_i); in DeterminantFromPivotedLUKernel()
126 typename TTypes<Scalar, 3>::ConstTensor lu_factor, in operator ()()
130 const int64 n = lu_factor.dimension(2); in operator ()()
136 config.virtual_thread_count, n, lu_factor.data(), pivots, nullptr, in operator ()()
149 typename TTypes<Scalar, 3>::ConstTensor lu_factor, in operator ()()
153 const int64 n = lu_factor.dimension(2); in operator ()()
158 config.virtual_thread_count, n, lu_factor.data(), pivots, sign.data(), in operator ()()
|
| D | determinant_op.h | 29 typename TTypes<Scalar, 3>::ConstTensor lu_factor,
39 typename TTypes<Scalar, 3>::ConstTensor lu_factor,
|
| /external/eigen/bench/btl/libs/gmm/ |
| D | gmm_interface.hh | 129 gmm::lu_factor(R, ipvt); in partial_lu_decomp()
|