Searched refs:kAccumulationIntegerBits (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | softmax.h | 36 static const int kAccumulationIntegerBits = 12; in Softmax() local
39 using FixedPointAccum = gemmlowp::FixedPoint<int32, kAccumulationIntegerBits>; in Softmax()
64 sum_of_exps = sum_of_exps + gemmlowp::Rescale<kAccumulationIntegerBits>( in Softmax()
71 sum_of_exps.raw(), kAccumulationIntegerBits, &num_bits_over_unit)); in Softmax()
|
| D | log_softmax.h | 38 static constexpr int kAccumulationIntegerBits = 12; in LogSoftmax() local
41 using F12 = gemmlowp::FixedPoint<int32, kAccumulationIntegerBits>; in LogSoftmax()
61 gemmlowp::Rescale<kAccumulationIntegerBits>( in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | softmax.h | 72 static const int kAccumulationIntegerBits = 12; in Softmax() local
75 using FixedPointAccum = gemmlowp::FixedPoint<int32, kAccumulationIntegerBits>; in Softmax()
100 sum_of_exps = sum_of_exps + gemmlowp::Rescale<kAccumulationIntegerBits>( in Softmax()
107 sum_of_exps.raw(), kAccumulationIntegerBits, &num_bits_over_unit)); in Softmax()
|
| D | reference_ops.h | 2614 static constexpr int kAccumulationIntegerBits = 12; in LogSoftmax() local
2618 using FixedPointAccum = gemmlowp::FixedPoint<int32, kAccumulationIntegerBits>; in LogSoftmax()
2642 sum_of_exps = sum_of_exps + gemmlowp::Rescale<kAccumulationIntegerBits>( in LogSoftmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4463 static const int kAccumulationIntegerBits = 12; in Softmax() local
4466 using FixedPointAccum = gemmlowp::FixedPoint<int32, kAccumulationIntegerBits>; in Softmax()
4513 gemmlowp::FixedPoint<int32x4_t, kAccumulationIntegerBits>; in Softmax()
4551 gemmlowp::Rescale<kAccumulationIntegerBits>( in Softmax()
4554 gemmlowp::Rescale<kAccumulationIntegerBits>( in Softmax()
4581 sum_of_exps + gemmlowp::Rescale<kAccumulationIntegerBits>( in Softmax()
4595 int num_bits_over_unit = kAccumulationIntegerBits - headroom_plus_one; in Softmax()
4715 static constexpr int kAccumulationIntegerBits = 12; in LogSoftmax() local
4719 using FixedPointAccum = gemmlowp::FixedPoint<int32, kAccumulationIntegerBits>; in LogSoftmax()
4744 sum_of_exps = sum_of_exps + gemmlowp::Rescale<kAccumulationIntegerBits>( in LogSoftmax()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4588 static const int kAccumulationIntegerBits = 12; in Softmax() local
4591 using FixedPointAccum = gemmlowp::FixedPoint<int32, kAccumulationIntegerBits>; in Softmax()
4638 gemmlowp::FixedPoint<int32x4_t, kAccumulationIntegerBits>; in Softmax()
4676 gemmlowp::Rescale<kAccumulationIntegerBits>( in Softmax()
4679 gemmlowp::Rescale<kAccumulationIntegerBits>( in Softmax()
4706 sum_of_exps + gemmlowp::Rescale<kAccumulationIntegerBits>( in Softmax()
4720 int num_bits_over_unit = kAccumulationIntegerBits - headroom_plus_one; in Softmax()
4975 static constexpr int kAccumulationIntegerBits = 12; in LogSoftmax() local
4979 using FixedPointAccum = gemmlowp::FixedPoint<int32, kAccumulationIntegerBits>; in LogSoftmax()
5005 sum_of_exps = sum_of_exps + gemmlowp::Rescale<kAccumulationIntegerBits>( in LogSoftmax()
|