diff options
| author |  Vitaly Lavrukhin <vlavrukhin@nvidia.com> | 2018-08-15 08:42:30 -0700 |
|---|---|---|
| committer | Vitaly Lavrukhin <vlavrukhin@nvidia.com> | 2018-08-15 08:42:30 -0700 |
| commit | e233afe9810e7a80b7b5d6e91fd87c5f47f2e72e (patch) | |
| tree | 38d8432e1214eb0cc6287e8a51df7ea0fc294c1b /tensorflow/core/util | |
| parent | 6d76eaaa751cc56a68dcf6e39c5d72191b9be26e (diff) | |
Switched to Eigen exp/log
Diffstat (limited to 'tensorflow/core/util')
| -rw-r--r-- | tensorflow/core/util/ctc/ctc_beam_search.h | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/core/util/ctc/ctc_beam_search.h b/tensorflow/core/util/ctc/ctc_beam_search.h index fa58b5cedf..b8390d3650 100644 --- a/tensorflow/core/util/ctc/ctc_beam_search.h +++ b/tensorflow/core/util/ctc/ctc_beam_search.h @@ -263,9 +263,9 @@ void CTCBeamSearchDecoder<CTCBeamState, CTCBeamComparer>::Step( // Get normalization term of softmax: log(sum(exp(logit[j]-max_coeff))). float logsumexp = 0.0; for (int j = 0; j < raw_input.size(); ++j) { - logsumexp += expf(raw_input(j) - max_coeff); + logsumexp += Eigen::numext::exp(raw_input(j) - max_coeff); } - logsumexp = logf(logsumexp); + logsumexp = Eigen::numext::log(logsumexp); // Final normalization offset to get correct log probabilities. float norm_offset = max_coeff + logsumexp; |