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.h115 FixedPoint0 shifted_scale = FixedPoint0::FromRaw(GetReciprocal( in Softmax() local
130 (shifted_scale * exp_in_0).raw(), in Softmax()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dcommon.h777 gemmlowp::FixedPoint<int32_t, 0> shifted_scale = in GetReciprocal() local
780 return shifted_scale.raw(); in GetReciprocal()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h4134 FixedPoint0 shifted_scale = FixedPoint0::FromRaw(GetReciprocal( in Softmax() local
4162 vqrdmulhq_n_s32(exp_0.raw(), shifted_scale.raw()), in Softmax()
4165 vqrdmulhq_n_s32(exp_1.raw(), shifted_scale.raw()), in Softmax()
4185 (shifted_scale * exp_in_0).raw(), num_bits_over_unit + 31 - 8); in Softmax()