Lines Matching refs:num_classes
94 int num_classes, const Vector& seq_len,
123 auto num_classes = inputs[0].cols();
136 if (inputs[t].cols() != num_classes) {
138 " to be: ", num_classes,
161 batch_size, num_classes, seq_len, labels, &max_u_prime, &l_primes);
167 auto ComputeLossAndGradients = [this, num_classes, &labels, &l_primes,
196 Matrix y(num_classes, seq_len(b));
206 // y, prob are in num_classes x seq_len(b)
262 max_seq_len * num_classes *
264 max_seq_len * 2 * (2 * num_classes + 1) *
267 ((2 * num_classes + 1) * cost_log_sum_exp +
268 num_classes * (cost_exp + Eigen::TensorOpCost::AddCost<float>()));
280 int batch_size, int num_classes, const Vector& seq_len,
308 if (label[i] >= num_classes - 1) {
315 "Saw a non-null label (index >= num_classes - 1) "
317 "null label, batch: ", b, " num_classes: ", num_classes,
330 } else if (l_i >= num_classes) {
332 "No label may be greater than num_classes. ",
333 "num_classes: ", num_classes, ", batch: ", b,
