Searched refs:qk_is_nonzero (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/client/lib/ |
| D | math.cc | 827 XlaOp qk_is_nonzero = Ne(qk, ScalarLike(qk, 0)); in IgammacContinuedFraction() local
830 t = Select(qk_is_nonzero, Abs((ans - r) / r), FullLike(t, 1)); in IgammacContinuedFraction()
831 ans = Select(qk_is_nonzero, r, ans); in IgammacContinuedFraction()
836 Select(qk_is_nonzero, (dpk_da - ans * dqk_da) / qk, dans_da); in IgammacContinuedFraction()
838 Select(qk_is_nonzero, Abs(dans_da_new - dans_da), FullLike(dans_da, 1)); in IgammacContinuedFraction()
|