Home
last modified time | relevance | path

Searched refs:shifted_scale (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dsoftmax.h114 FixedPoint0 shifted_scale = FixedPoint0::FromRaw(GetReciprocal( in Softmax() local
129 (shifted_scale * exp_in_0).raw(), in Softmax()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h634 gemmlowp::FixedPoint<int32_t, 0> shifted_scale = in GetReciprocal() local
637 return shifted_scale.raw(); in GetReciprocal()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h4129 FixedPoint0 shifted_scale = FixedPoint0::FromRaw(GetReciprocal( in Softmax() local
4157 vqrdmulhq_n_s32(exp_0.raw(), shifted_scale.raw()), in Softmax()
4160 vqrdmulhq_n_s32(exp_1.raw(), shifted_scale.raw()), in Softmax()
4180 (shifted_scale * exp_in_0).raw(), num_bits_over_unit + 31 - 8); in Softmax()