Searched refs:max_coeff (Results 1 – 2 of 2) sorted by relevance
253 float max_coeff; in Step() local255 max_coeff = GetTopK(label_selection_size_, raw_input, &top_k_logits, in Step()258 max_coeff = raw_input.maxCoeff(); in Step()264 logsumexp += Eigen::numext::exp(raw_input(j) - max_coeff); in Step()268 float norm_offset = max_coeff + logsumexp; in Step()271 (label_selection_margin_ >= 0) ? (max_coeff - label_selection_margin_) in Step()
1460 const int max_coeff = (MAX_COEFF_THRESH << 16) + MAX_COEFF_THRESH; in CollectHistogram_MIPSdspR2() local1474 : [dist]"r"(distribution), [out]"r"(out), [max_coeff]"r"(max_coeff) in CollectHistogram_MIPSdspR2()
Completed in 10 milliseconds