Home
last modified time | relevance | path

Searched refs:input_multiplier (Results 1 – 16 of 16) sorted by relevance

/external/tensorflow/tensorflow/lite/experimental/micro/kernels/
Dsoftmax.cc32 int32_t input_multiplier = 0; member
51 &data->input_multiplier, &data->input_left_shift); in CalculateSoftmaxOpData()
97 op_params.input_multiplier = data->input_multiplier; in Softmax1DQuantized()
116 op_params.input_multiplier = data->input_multiplier; in Softmax2DQuantized()
137 op_params.input_multiplier = data->input_multiplier; in Softmax4DQuantized()
/external/tensorflow/tensorflow/lite/kernels/
Dactivations.cc46 int32_t input_multiplier = 0; member
132 &data->input_multiplier, in TanhPrepare()
201 &data->input_multiplier, in SigmoidPrepare()
256 &data->input_multiplier, &data->input_left_shift); in SoftmaxPrepare()
287 &data->input_multiplier, &data->input_left_shift, in LogSoftmaxPrepare()
452 params.input_multiplier = data->input_multiplier; in TanhEval()
471 data->input_multiplier, data->input_left_shift, size, in TanhEval()
519 params.input_multiplier = data->input_multiplier; in SigmoidEval()
537 data->input_multiplier, data->input_left_shift, input_size, in SigmoidEval()
618 op_params.input_multiplier = data->input_multiplier; in Softmax1DQuantizedUint8()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/
Dlog_softmax.h23 inline void LogSoftmax(int32_t input_multiplier, int32_t input_shift, in LogSoftmax() argument
58 input_diff, input_multiplier, input_shift); in LogSoftmax()
87 input_diff, input_multiplier, input_shift); in LogSoftmax()
Dlogistic.h25 int32_t input_multiplier, int32_t input_left_shift, in Logistic() argument
44 input, input_multiplier, input_left_shift); in Logistic()
Dtanh.h25 int32_t input_multiplier, int32_t input_shift, in Tanh() argument
44 MultiplyByQuantizedMultiplier(input, input_multiplier, input_shift); in Tanh()
Dsoftmax.h27 const int32 input_beta_multiplier = params.input_multiplier; in Softmax()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h1447 params.input_multiplier = input_beta_multiplier; in Softmax()
1475 int32 input_multiplier, int32 input_left_shift, in LogSoftmax() argument
1480 params.input_multiplier = input_multiplier; in LogSoftmax()
1489 int32 input_multiplier, int32 input_left_shift, in LogSoftmax() argument
1493 LogSoftmax(input_data, DimsToShape(input_dims), input_multiplier, in LogSoftmax()
1501 int32 input_multiplier, int input_left_shift, in Logistic() argument
1506 params.input_multiplier = input_multiplier; in Logistic()
1519 int32 input_multiplier, int input_left_shift, in Logistic() argument
1522 input_range_radius, input_multiplier, input_left_shift, output_data, in Logistic()
1554 int32 input_multiplier, int input_left_shift, in Tanh() argument
[all …]
Doptimized_ops.h4454 const int32 input_beta_multiplier = params.input_multiplier; in Softmax()
4704 const int32 input_multiplier = params.input_multiplier; in LogSoftmax() local
4741 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
4772 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
4813 const int32 input_multiplier = params.input_multiplier; in Logistic() local
4868 vqrdmulhq_n_s32(input_val_rescaled_0, input_multiplier); in Logistic()
4870 vqrdmulhq_n_s32(input_val_rescaled_1, input_multiplier); in Logistic()
4872 vqrdmulhq_n_s32(input_val_rescaled_2, input_multiplier); in Logistic()
4874 vqrdmulhq_n_s32(input_val_rescaled_3, input_multiplier); in Logistic()
4929 input_val_centered, input_multiplier, input_left_shift); in Logistic()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h711 params.input_multiplier = input_beta_multiplier; in Softmax()
725 int32 input_multiplier, int32 input_left_shift, in LogSoftmax() argument
730 params.input_multiplier = input_multiplier; in LogSoftmax()
740 int32 input_multiplier, int input_left_shift, in Logistic() argument
745 params.input_multiplier = input_multiplier; in Logistic()
759 int32 input_multiplier, int input_left_shift, in Tanh() argument
764 params.input_multiplier = input_multiplier; in Tanh()
1700 int32 input_multiplier, int32 input_left_shift, in LogSoftmax() argument
1704 LogSoftmax(input_data, DimsToShape(input_dims), input_multiplier, in LogSoftmax()
1718 int32 input_multiplier, int input_left_shift, in Logistic() argument
[all …]
Dsoftmax.h63 const int32 input_beta_multiplier = params.input_multiplier; in Softmax()
Dreference_ops.h2603 const int32 input_multiplier = params.input_multiplier; in LogSoftmax() local
2639 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
2671 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
2713 const int32 input_multiplier = params.input_multiplier; in Logistic() local
2729 input_val_centered, input_multiplier, input_left_shift); in Logistic()
2793 const int32 input_multiplier = params.input_multiplier; in Tanh() local
2810 input_val_centered, input_multiplier, input_left_shift); in Tanh()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/
Doptimized_ops.h4579 const int32 input_beta_multiplier = params.input_multiplier; in Softmax()
4794 params.input_multiplier = input_beta_multiplier; in Softmax()
4964 const int32 input_multiplier = params.input_multiplier; in LogSoftmax() local
5002 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
5033 input_diff, input_multiplier, input_left_shift); in LogSoftmax()
5053 int32 input_multiplier, int32 input_left_shift, in LogSoftmax() argument
5058 params.input_multiplier = input_multiplier; in LogSoftmax()
5090 const int32 input_multiplier = params.input_multiplier; in Logistic() local
5145 vqrdmulhq_n_s32(input_val_rescaled_0, input_multiplier); in Logistic()
5147 vqrdmulhq_n_s32(input_val_rescaled_1, input_multiplier); in Logistic()
[all …]
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h851 int32 input_multiplier; member
911 int32 input_multiplier; member
958 int32 input_multiplier; member
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h874 int32 input_multiplier; member
956 int32 input_multiplier; member
1003 int32 input_multiplier; member
Dsoftmax_quantized_test.cc124 params.input_multiplier = input_beta_multiplier; in RunOneSoftmaxTest()
Dlogsoftmax_quantized_test.cc174 params.input_multiplier = input_beta_multiplier; in RunOneLogSoftmaxTest()