Home
last modified time | relevance | path

Searched refs:reverse_scaling_right_shift (Results 1 – 9 of 9) 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()
177 params.reverse_scaling_right_shift = reverse_scaling_right_shift; in RunOneLogSoftmaxTest()
212 int reverse_scaling_right_shift; in RunOneLogSoftmaxTest() local
217 &reverse_scaling_right_shift); in RunOneLogSoftmaxTest()
218 reverse_scaling_right_shift *= -1; in RunOneLogSoftmaxTest()
229 reverse_scaling_right_shift, diff_min, outer_size, inner_size, input_data, in RunOneLogSoftmaxTest()
Dtypes.h960 int32 reverse_scaling_right_shift; member
/external/tensorflow/tensorflow/lite/kernels/
Dactivations.cc54 int32_t reverse_scaling_right_shift = 0; member
288 &data->reverse_scaling_divisor, &data->reverse_scaling_right_shift); in LogSoftmaxPrepare()
289 data->reverse_scaling_right_shift *= -1; in LogSoftmaxPrepare()
850 op_params.reverse_scaling_right_shift = data->reverse_scaling_right_shift; in LogSoftmaxEval()
873 data->reverse_scaling_divisor, data->reverse_scaling_right_shift, in LogSoftmaxEval()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1477 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
1483 params.reverse_scaling_right_shift = reverse_scaling_right_shift; in LogSoftmax()
1491 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
1495 reverse_scaling_right_shift, diff_min, output_data, in LogSoftmax()
Doptimized_ops.h4707 const int32 reverse_scaling_right_shift = params.reverse_scaling_right_shift; in LogSoftmax() local
4765 -reverse_scaling_right_shift)); in LogSoftmax()
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h727 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
733 params.reverse_scaling_right_shift = reverse_scaling_right_shift; in LogSoftmax()
1702 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
1706 reverse_scaling_right_shift, diff_min, output_data, in LogSoftmax()
Dreference_ops.h2606 const int32 reverse_scaling_right_shift = params.reverse_scaling_right_shift; in LogSoftmax() local
2663 -reverse_scaling_right_shift)); in LogSoftmax()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h915 int32 reverse_scaling_right_shift; member
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h4967 const int32 reverse_scaling_right_shift = params.reverse_scaling_right_shift; in LogSoftmax() local
5026 -reverse_scaling_right_shift)); in LogSoftmax()
5055 int32 reverse_scaling_right_shift, int diff_min, in LogSoftmax() argument
5061 params.reverse_scaling_right_shift = reverse_scaling_right_shift; in LogSoftmax()