Searched refs:logsumexp (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | scan_ops.h | 58 auto logsumexp = add(log1p(exp(sub(mi, ma))), ma); in operator() local
59 return cmp_lt(ma, Eigen::NumTraits<T>::lowest()) ? ma : logsumexp; in operator()
72 auto logsumexp = padd(plog1p(pexp(psub(mi, ma))), ma); in packetOp() local
74 logsumexp); in packetOp()
81 LogSumExp<T> logsumexp; in reduce() local
82 *accum = logsumexp(*accum, t); in reduce()
88 LogSumExp<T> logsumexp; in reducePacket() local
89 *accum = logsumexp.packetOp(*accum, p); in reducePacket()
132 auto logsumexp = add(log(sum_reducer.finalizeBoth( in finalizeBoth() local
135 return cmp_lt(ma, Eigen::NumTraits<T>::lowest()) ? initialize() : logsumexp; in finalizeBoth()
|
| /external/tensorflow/tensorflow/lite/experimental/kernels/ |
| D | ctc_beam_search.h | 262 float logsumexp = 0.0; in Step() local
264 logsumexp += Eigen::numext::exp(raw_input(j) - max_coeff); in Step()
266 logsumexp = Eigen::numext::log(logsumexp); in Step()
268 float norm_offset = max_coeff + logsumexp; in Step()
|
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_beam_search.h | 267 T logsumexp = T(0.0); in Step() local
269 logsumexp += Eigen::numext::exp(raw_input(j) - max_coeff); in Step()
271 logsumexp = Eigen::numext::log(logsumexp); in Step()
273 T norm_offset = max_coeff + logsumexp; in Step()
|
| /external/llvm-project/mlir/docs/Dialects/ |
| D | Vector.md | 220 `max/+`, `or/and`, `logsumexp/+` ...) . In other words, by operating on a
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | backend.py | 2608 def logsumexp(x, axis=None, keepdims=False): function
|