Home
last modified time | relevance | path

Searched refs:sum_log_abs_det (Results 1 – 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddeterminant_op_gpu.cu.cc102 RealScalar sum_log_abs_det = RealScalar(0); in DeterminantFromPivotedLUKernel() local
106 sum_log_abs_det += Eigen::numext::log(abs_i); in DeterminantFromPivotedLUKernel()
109 if (!Eigen::numext::isfinite(sum_log_abs_det)) { in DeterminantFromPivotedLUKernel()
111 sum_log_abs_det = sum_log_abs_det > 0 ? -Eigen::numext::log(RealScalar(0)) in DeterminantFromPivotedLUKernel()
116 log_abs_det[o_idx] = Scalar(sum_log_abs_det); in DeterminantFromPivotedLUKernel()
118 log_abs_det[o_idx] = prod_sign * Eigen::numext::exp(sum_log_abs_det); in DeterminantFromPivotedLUKernel()