Searched refs:l_prime (Results 1 – 2 of 2) sorted by relevance
/external/tensorflow/tensorflow/core/util/ctc/ |
D | ctc_loss_calculator.cc | 25 const std::vector<int>& l_prime, const Matrix& y, bool ctc_merge_repeated, in CalculateForwardVariables() argument 31 int U = l_prime.size(); in CalculateForwardVariables() 39 auto label_0 = (l_prime.size() > 1) ? l_prime[1] : blank_index_; in CalculateForwardVariables() 51 if (ctc_merge_repeated || l_prime[u] == blank_index_) { in CalculateForwardVariables() 64 ctc_merge_repeated && (l_prime[u] == l_prime[u - 2]); in CalculateForwardVariables() 65 if (l_prime[u] != blank_index_ && !matching_labels_merge) { in CalculateForwardVariables() 72 log(y(l_prime[u], output_delay_ + t)) + sum_log_alpha; in CalculateForwardVariables() 79 const std::vector<int>& l_prime, const Matrix& y, bool ctc_merge_repeated, in CalculateBackwardVariables() argument 87 int U = l_prime.size(); in CalculateBackwardVariables() 101 if (ctc_merge_repeated || l_prime[u] == blank_index_) { in CalculateBackwardVariables() [all …]
|
D | ctc_loss_calculator.h | 72 void CalculateForwardVariables(const std::vector<int>& l_prime, 76 void CalculateBackwardVariables(const std::vector<int>& l_prime, 80 void CalculateGradient(const std::vector<int>& l_prime, const Matrix& y, 85 std::vector<int>* l_prime) const; 190 const std::vector<int>& l_prime = l_primes[b]; in CalculateLoss() local 192 Matrix log_alpha_b(l_prime.size(), seq_len(b) - this->output_delay_); in CalculateLoss() 193 Matrix log_beta_b(l_prime.size(), seq_len(b) - this->output_delay_); in CalculateLoss() 219 CalculateForwardVariables(l_prime, y_b, ctc_merge_repeated, &log_alpha_b); in CalculateLoss() 221 CalculateBackwardVariables(l_prime, y_b, ctc_merge_repeated, &log_beta_b); in CalculateLoss() 226 for (int u = 0; u < l_prime.size(); ++u) { in CalculateLoss() [all …]
|