Home
last modified time | relevance | path

Searched refs:sum_of_exps (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsoftmax.h98 FixedPointAccum sum_of_exps = FixedPointAccum::Zero(); in Softmax() local
108 sum_of_exps = sum_of_exps + gemmlowp::Rescale<kAccumulationIntegerBits>( in Softmax()
115 sum_of_exps.raw(), kAccumulationIntegerBits, &num_bits_over_unit)); in Softmax()
187 int32_t sum_of_exps = 0; // Q16.15 fixed point format. in SoftmaxInt16() local
192 sum_of_exps += exp_results_Q015[c]; in SoftmaxInt16()
197 CountLeadingZeros(static_cast<uint32_t>(sum_of_exps)); in SoftmaxInt16()
199 ((static_cast<int64_t>(sum_of_exps) << (headroom_plus_one - 1)) + in SoftmaxInt16()
Dreference_ops.h970 FixedPointAccum sum_of_exps = FixedPointAccum::Zero(); in LogSoftmax() local
980 sum_of_exps = sum_of_exps + gemmlowp::Rescale<kAccumulationIntegerBits>( in LogSoftmax()
987 sum_of_exps) in LogSoftmax()
/external/tensorflow/tensorflow/lite/micro/kernels/xtensa/
Dsoftmax.cc80 uint32_t sum_of_exps = 0; in SoftmaxHifimini() local
85 sum_of_exps += in SoftmaxHifimini()
91 TFLITE_DCHECK(sum_of_exps >= kMaxExponentValue); in SoftmaxHifimini()
101 scaled_output / sum_of_exps + std::numeric_limits<int16_t>::min(); in SoftmaxHifimini()
103 uint32_t remainder = scaled_output % sum_of_exps; in SoftmaxHifimini()
104 if (remainder * 2 >= sum_of_exps) { in SoftmaxHifimini()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h4063 FixedPointAccum sum_of_exps = FixedPointAccum::Zero(); in Softmax() local
4108 sum_of_exps = in Softmax()
4119 sum_of_exps = in Softmax()
4120 sum_of_exps + gemmlowp::Rescale<kAccumulationIntegerBits>( in Softmax()
4130 sum_of_exps.raw(), kAccumulationIntegerBits, &num_bits_over_unit)); in Softmax()