commit | 2a886bdd1b70b27b8cf85b239c96518812c3b4ed | [log] [tgz] |
---|---|---|
author | Treehugger Robot <treehugger-gerrit@google.com> | Wed Jun 12 23:59:19 2019 +0000 |
committer | Gerrit Code Review <noreply-gerritcodereview@google.com> | Wed Jun 12 23:59:19 2019 +0000 |
tree | 553a8cdc135f70197f7aeadce995db415ab7282b | |
parent | 0e556ab7ea52c319cc41e75a0076f6357381d86a [diff] | |
parent | 2d33b3f144ba2365b620cbe9dcf18f51d594e39a [diff] |
Merge "Fix PRELU underflow/overflow issue."
diff --git a/nn/common/operations/PRelu.cpp b/nn/common/operations/PRelu.cpp index 0bf4ce6..68d1e5b 100644 --- a/nn/common/operations/PRelu.cpp +++ b/nn/common/operations/PRelu.cpp
@@ -86,6 +86,7 @@ tflite::MultiplyByQuantizedMultiplier( input * alpha, output_multiplier_neg, output_shift_neg); } + output_val = std::max(0, std::min(255, output_val)); return static_cast<uint8_t>(output_val); }, aData, aShape, bData, bShape, outputData, outputShape);