experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #26384 from luxupu:patch-3

Browse Source 
PiperOrigin-RevId: 238310428
This commit is contained in:
TensorFlower Gardener 2019-03-13 14:39:07 -07:00
commit d564861978
1 changed files with 2 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
tensorflow/lite/kernels/internal/reference/integer_ops/softmax.h
View File

@ -85,9 +85,8 @@ inline void Softmax(const SoftmaxParams& params,
(shifted_scale * exp_in_0).raw(), num_bits_over_unit + 31 - 8); (shifted_scale * exp_in_0).raw(), num_bits_over_unit + 31 - 8);
const int32 shifted_output = unsat_output - 128; const int32 shifted_output = unsat_output - 128;
output_data[i * depth + c] = static_cast<int8>( output_data[i * depth + c] =
std::max(std::min(shifted_output, static_cast<int32>(127)), static_cast<int8>(std::max(std::min(shifted_output, 127), -128));
static_cast<int32>(-128)));
} else { } else {
output_data[i * depth + c] = -128; output_data[i * depth + c] = -128;