Searched refs:log_sum (Results 1 – 3 of 3) sorted by relevance
2592 const float log_sum = std::log(sum); in LogSoftmax() local2594 output_data[i * depth + c] = input_data[i * depth + c] - max - log_sum; in LogSoftmax()
4830 const float log_sum = std::log(sum); in LogSoftmax() local4832 block_output_data[c] = block_input_data[c] - max - log_sum; in LogSoftmax()
4692 const float log_sum = std::log(sum); in LogSoftmax() local4694 block_output_data[c] = block_input_data[c] - max - log_sum; in LogSoftmax()