Searched refs:PreprocessSoftmaxScaling (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | quantization_util.cc | 254 void PreprocessSoftmaxScaling(double beta, double input_scale, in PreprocessSoftmaxScaling() function
291 PreprocessSoftmaxScaling(beta, input_scale, input_integer_bits, in PreprocessLogSoftmaxScalingExp()
|
| D | quantization_util.h | 240 void PreprocessSoftmaxScaling(double beta, double input_scale,
|
| D | quantization_util_test.cc | 384 TEST(QuantizationUtilTest, PreprocessSoftmaxScaling) { in TEST() argument
388 PreprocessSoftmaxScaling(beta, scale, integer_bits, &q, &s); in TEST()
|
| D | softmax_quantized_test.cc | 115 tflite::PreprocessSoftmaxScaling(beta, input_scale, kScaledDiffIntegerBits, in RunOneSoftmaxTest()
|
| /external/tensorflow/tensorflow/lite/experimental/micro/kernels/ |
| D | softmax.cc | 49 tflite::PreprocessSoftmaxScaling( in CalculateSoftmaxOpData()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | activations.cc | 254 tflite::PreprocessSoftmaxScaling( in SoftmaxPrepare()
|