Searched refs:reduce_weighted_logsumexp (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/python/kernel_tests/distributions/ |
| D | util_test.py | 829 actual, actual_sgn = du.reduce_weighted_logsumexp(
851 actual, actual_sgn = du.reduce_weighted_logsumexp(
869 actual, actual_sgn = du.reduce_weighted_logsumexp(
886 np.log(4), self.evaluate(du.reduce_weighted_logsumexp(x, w)))
891 self.evaluate(du.reduce_weighted_logsumexp(x, w, axis=0)))
895 self.evaluate(du.reduce_weighted_logsumexp(x, w, axis=1)))
900 du.reduce_weighted_logsumexp(x, w, axis=1, keep_dims=True)))
904 self.evaluate(du.reduce_weighted_logsumexp(x, w, axis=[0, 1])))
|
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 1050 def reduce_weighted_logsumexp(logx, function
|