Home
last modified time | relevance | path

Searched refs:reverse_scaling_divisor (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dlogsoftmax_quantized_test.cc160 int32 reverse_scaling_divisor; in RunOneLogSoftmaxTest() local
165 &input_beta_left_shift, &reverse_scaling_divisor, in RunOneLogSoftmaxTest()
177 params.reverse_scaling_divisor = reverse_scaling_divisor; in RunOneLogSoftmaxTest()
217 int32 reverse_scaling_divisor; in RunOneLogSoftmaxTest() local
222 &input_beta_left_shift, &reverse_scaling_divisor, in RunOneLogSoftmaxTest()
234 input_beta_multiplier, input_beta_left_shift, reverse_scaling_divisor, in RunOneLogSoftmaxTest()
Dquantization_util.cc325 int32_t* reverse_scaling_divisor, in PreprocessLogSoftmaxScalingExp() argument
334 reverse_scaling_divisor, in PreprocessLogSoftmaxScalingExp()
Dquantization_util.h248 int32_t* reverse_scaling_divisor,
Dtypes.h934 int32_t reverse_scaling_divisor; member
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlog_softmax.h66 const int32_t reverse_scaling_divisor = params.reverse_scaling_divisor; in LogSoftmax() local
126 rescaled_diff_min, reverse_scaling_divisor, in LogSoftmax()
162 const int32_t reverse_scaling_divisor = params.reverse_scaling_divisor; in LogSoftmaxQuantized() local
214 reverse_scaling_divisor, in LogSoftmaxQuantized()
Dlegacy_reference_ops.h778 int32 reverse_scaling_divisor, in LogSoftmax() argument
784 params.reverse_scaling_divisor = reverse_scaling_divisor; in LogSoftmax()
1799 int32 reverse_scaling_divisor, in LogSoftmax() argument
1803 input_left_shift, reverse_scaling_divisor, in LogSoftmax()
/external/tensorflow/tensorflow/lite/kernels/
Dactivations.cc90 int32_t reverse_scaling_divisor = 0; member
689 &input_left_shift, &data->params.reverse_scaling_divisor, in LogSoftmaxPrepare()
1320 op_params.reverse_scaling_divisor, in LogSoftmaxEval()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h4246 int32 reverse_scaling_divisor, in LogSoftmax() argument
4252 params.reverse_scaling_divisor = reverse_scaling_divisor; in LogSoftmax()
4261 int32 reverse_scaling_divisor, in LogSoftmax() argument
4266 reverse_scaling_divisor, reverse_scaling_right_shift, diff_min, in LogSoftmax()