experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Qualify calls to some functions from <cmath>.

Browse Source 
PiperOrigin-RevId: 242057286
This commit is contained in:
A. Unique TensorFlower 2019-04-04 19:40:01 -07:00 committed by TensorFlower Gardener
parent 82735fd7a6
commit 3bf4628d75
1 changed files with 7 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

13
tensorflow/core/util/ctc/ctc_loss_calculator.cc
View File

@ -14,6 +14,7 @@ limitations under the License.
==============================================================================*/ ==============================================================================*/
#include "tensorflow/core/util/ctc/ctc_loss_calculator.h" #include "tensorflow/core/util/ctc/ctc_loss_calculator.h"
#include <cmath>
namespace tensorflow { namespace tensorflow {
namespace ctc { namespace ctc {
@ -34,10 +35,10 @@ void CTCLossCalculator::CalculateForwardVariables(
CHECK_EQ(U, log_alpha->rows()); CHECK_EQ(U, log_alpha->rows());
// Initial alpha values in (GravesTh) Eq 7.5 and Eq 7.6. // Initial alpha values in (GravesTh) Eq 7.5 and Eq 7.6.
log_alpha->coeffRef(0, 0) = log(y(blank_index_, output_delay_)); log_alpha->coeffRef(0, 0) = std::log(y(blank_index_, output_delay_));
// Below, l_prime[1] == labels[0] // Below, l_prime[1] == labels[0]
auto label_0 = (l_prime.size() > 1) ? l_prime[1] : blank_index_; auto label_0 = (l_prime.size() > 1) ? l_prime[1] : blank_index_;
log_alpha->coeffRef(1, 0) = log(y(label_0, output_delay_)); log_alpha->coeffRef(1, 0) = std::log(y(label_0, output_delay_));
for (int t = 1; t < T; ++t) { for (int t = 1; t < T; ++t) {
// If there is not enough time to output the remaining labels or // If there is not enough time to output the remaining labels or
@ -69,7 +70,7 @@ void CTCLossCalculator::CalculateForwardVariables(
} }
// Multiply the summed alphas with the activation log probability. // Multiply the summed alphas with the activation log probability.
log_alpha->coeffRef(u, t) = log_alpha->coeffRef(u, t) =
log(y(l_prime[u], output_delay_ + t)) + sum_log_alpha; std::log(y(l_prime[u], output_delay_ + t)) + sum_log_alpha;
} // End (GravesTh) Eq 7.9. } // End (GravesTh) Eq 7.9.
} }
} }
@ -102,7 +103,7 @@ void CTCLossCalculator::CalculateBackwardVariables(
log_beta->coeffRef(u, t) = log_beta->coeffRef(u, t) =
LogSumExp(log_beta->coeff(u, t), LogSumExp(log_beta->coeff(u, t),
log_beta->coeff(u, t + 1) + log_beta->coeff(u, t + 1) +
log(y(l_prime[u], output_delay_ + t + 1))); std::log(y(l_prime[u], output_delay_ + t + 1)));
} }
// Add in the u + 1, t + 1 term. // Add in the u + 1, t + 1 term.
@ -110,7 +111,7 @@ void CTCLossCalculator::CalculateBackwardVariables(
log_beta->coeffRef(u, t) = log_beta->coeffRef(u, t) =
LogSumExp(log_beta->coeff(u, t), LogSumExp(log_beta->coeff(u, t),
log_beta->coeff(u + 1, t + 1) + log_beta->coeff(u + 1, t + 1) +
log(y(l_prime[u + 1], output_delay_ + t + 1))); std::log(y(l_prime[u + 1], output_delay_ + t + 1)));
} }
// Add in the u + 2, t + 1 term if l_prime(u) != blank or l_prime(u+2). // Add in the u + 2, t + 1 term if l_prime(u) != blank or l_prime(u+2).
@ -122,7 +123,7 @@ void CTCLossCalculator::CalculateBackwardVariables(
log_beta->coeffRef(u, t) = log_beta->coeffRef(u, t) =
LogSumExp(log_beta->coeff(u, t), LogSumExp(log_beta->coeff(u, t),
log_beta->coeff(u + 2, t + 1) + log_beta->coeff(u + 2, t + 1) +
log(y(l_prime[u + 2], output_delay_ + t + 1))); std::log(y(l_prime[u + 2], output_delay_ + t + 1)));
} }
} // End (GravesTh) Eq. 7.15 } // End (GravesTh) Eq. 7.15
} }