Searched refs:prob_quantized (Results 1 – 1 of 1) sorted by relevance
3793 const int32_t prob_quantized = in Softmax() local3796 std::max(std::min(clamp_max, prob_quantized), clamp_min)); in Softmax()4162 const int32_t prob_quantized = std::rint(log_prob) + params.zero_point; in LogSoftmax() local4164 std::max(std::min(clamp_max, prob_quantized), clamp_min)); in LogSoftmax()