| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | logsoftmax_quantized_test.cc | 158 int32 input_beta_multiplier; in RunOneLogSoftmaxTest() local
164 beta, input_scale, kScaledDiffIntegerBits, &input_beta_multiplier, in RunOneLogSoftmaxTest()
174 params.input_multiplier = input_beta_multiplier; in RunOneLogSoftmaxTest()
209 int32 input_beta_multiplier; in RunOneLogSoftmaxTest() local
215 beta, input_scale, kScaledDiffIntegerBits, &input_beta_multiplier, in RunOneLogSoftmaxTest()
228 input_beta_multiplier, input_beta_left_shift, reverse_scaling_divisor, in RunOneLogSoftmaxTest()
|
| D | softmax_quantized_test.cc | 112 int32 input_beta_multiplier; in RunOneSoftmaxTest() local
116 &input_beta_multiplier, in RunOneSoftmaxTest()
124 params.input_multiplier = input_beta_multiplier; in RunOneSoftmaxTest()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/integer_ops/ |
| D | softmax.h | 27 const int32 input_beta_multiplier = params.input_multiplier; in Softmax() local
61 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
79 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | softmax.h | 63 const int32 input_beta_multiplier = params.input_multiplier; in Softmax() local
97 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
115 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
|
| D | legacy_reference_ops.h | 707 int32 input_beta_multiplier, int32 input_beta_left_shift, in Softmax() argument
711 params.input_multiplier = input_beta_multiplier; in Softmax()
1685 int32 input_beta_multiplier, int32 input_beta_left_shift, in Softmax() argument
1688 Softmax(input_data, DimsToShape(input_dims), input_beta_multiplier, in Softmax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | legacy_optimized_ops.h | 1443 int32 input_beta_multiplier, int32 input_beta_left_shift, in Softmax() argument
1447 params.input_multiplier = input_beta_multiplier; in Softmax()
1453 int32 input_beta_multiplier, int32 input_beta_left_shift, in Softmax() argument
1456 Softmax(input_data, DimsToShape(input_dims), input_beta_multiplier, in Softmax()
|
| D | optimized_ops.h | 4454 const int32 input_beta_multiplier = params.input_multiplier; in Softmax() local
4518 FixedPoint0Int32x4::FromScalarRaw(input_beta_multiplier); in Softmax()
4577 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
4644 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 4579 const int32 input_beta_multiplier = params.input_multiplier; in Softmax() local
4643 FixedPoint0Int32x4::FromScalarRaw(input_beta_multiplier); in Softmax()
4702 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
4769 input_diff, input_beta_multiplier, input_beta_left_shift); in Softmax()
4790 int32 input_beta_multiplier, int32 input_beta_left_shift, in Softmax() argument
4794 params.input_multiplier = input_beta_multiplier; in Softmax()
|