Searched refs:log_alpha (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ctc/ |
| D | ctc_loss_calculator.cc | 26 Matrix* log_alpha) const { in CalculateForwardVariables()
29 log_alpha->setConstant(kLogZero); in CalculateForwardVariables()
32 int T = log_alpha->cols(); in CalculateForwardVariables()
34 CHECK_EQ(U, log_alpha->rows()); in CalculateForwardVariables()
37 log_alpha->coeffRef(0, 0) = log(y(blank_index_, output_delay_)); in CalculateForwardVariables()
40 log_alpha->coeffRef(1, 0) = log(y(label_0, output_delay_)); in CalculateForwardVariables()
52 sum_log_alpha = log_alpha->coeff(u, t - 1); in CalculateForwardVariables()
58 LogSumExp(sum_log_alpha, log_alpha->coeff(u - 1, t - 1)); in CalculateForwardVariables()
67 LogSumExp(sum_log_alpha, log_alpha->coeff(u - 2, t - 1)); in CalculateForwardVariables()
71 log_alpha->coeffRef(u, t) = in CalculateForwardVariables()
[all …]
|
| D | ctc_loss_calculator.h | 74 Matrix* log_alpha) const;
81 const Matrix& log_alpha, const Matrix& log_beta,
|