Home
last modified time | relevance | path

Searched refs:reverse_scaling_right_shift (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/internal/
Dlogsoftmax_quantized_test.cc161 int reverse_scaling_right_shift; in RunOneLogSoftmaxTest() local
166 &reverse_scaling_right_shift); in RunOneLogSoftmaxTest()
167 reverse_scaling_right_shift *= -1; in RunOneLogSoftmaxTest()
178 params.reverse_scaling_right_shift = reverse_scaling_right_shift; in RunOneLogSoftmaxTest()
218 int reverse_scaling_right_shift; in RunOneLogSoftmaxTest() local
223 &reverse_scaling_right_shift); in RunOneLogSoftmaxTest()
224 reverse_scaling_right_shift *= -1; in RunOneLogSoftmaxTest()
235 reverse_scaling_right_shift, diff_min, outer_size, inner_size, input_data, in RunOneLogSoftmaxTest()
Dtypes.h935 int32_t reverse_scaling_right_shift; member
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlog_softmax.h67 const int32_t reverse_scaling_right_shift = in LogSoftmax() local
68 params.reverse_scaling_right_shift; in LogSoftmax()
127 -reverse_scaling_right_shift)); in LogSoftmax()
163 const int32_t reverse_scaling_right_shift = in LogSoftmaxQuantized() local
164 params.reverse_scaling_right_shift; in LogSoftmaxQuantized()
215 -reverse_scaling_right_shift)); in LogSoftmaxQuantized()
Dlegacy_reference_ops.h779 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
785 params.reverse_scaling_right_shift = reverse_scaling_right_shift; in LogSoftmax()
1800 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
1804 reverse_scaling_right_shift, diff_min, output_data, in LogSoftmax()
/external/tensorflow/tensorflow/lite/kernels/
Dactivations.cc91 int32_t reverse_scaling_right_shift = 0; member
685 int reverse_scaling_right_shift; in LogSoftmaxPrepare() local
690 &reverse_scaling_right_shift); in LogSoftmaxPrepare()
691 reverse_scaling_right_shift *= -1; in LogSoftmaxPrepare()
693 data->params.reverse_scaling_right_shift = reverse_scaling_right_shift; in LogSoftmaxPrepare()
1321 op_params.reverse_scaling_right_shift, op_params.diff_min, in LogSoftmaxEval()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h4247 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
4253 params.reverse_scaling_right_shift = reverse_scaling_right_shift; in LogSoftmax()
4262 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
4266 reverse_scaling_divisor, reverse_scaling_right_shift, diff_min, in LogSoftmax()