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

Merge pull request #35302 from aabadie:pr/lite/micro/use_global_round

Browse Source 
PiperOrigin-RevId: 300831411
Change-Id: I40452cc7899f9e587e58fd09357de74f7ea2d8bb
This commit is contained in:
TensorFlower Gardener 2020-03-13 14:55:13 -07:00
commit 8e80c097df
1 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/lite/kernels/internal/reference/concatenation.h
View File

@ -18,6 +18,7 @@ limitations under the License.
#include "tensorflow/lite/kernels/internal/common.h" #include "tensorflow/lite/kernels/internal/common.h"
#include "tensorflow/lite/kernels/internal/compatibility.h" #include "tensorflow/lite/kernels/internal/compatibility.h"
#include "tensorflow/lite/kernels/internal/round.h"
#include "tensorflow/lite/kernels/internal/types.h" #include "tensorflow/lite/kernels/internal/types.h"
namespace tflite { namespace tflite {
@ -121,9 +122,9 @@ inline void ConcatenationWithScaling(const ConcatenationParams& params,
const float scale = input_scale[i] * inverse_output_scale; const float scale = input_scale[i] * inverse_output_scale;
const float bias = -input_zeropoint[i] * scale; const float bias = -input_zeropoint[i] * scale;
for (int j = 0; j < copy_size; ++j) { for (int j = 0; j < copy_size; ++j) {
const int32_t value = const int32_t value = static_cast<int32_t>(tflite::TfLiteRound(
static_cast<int32_t>(std::round(input_ptr[j] * scale + bias)) + input_ptr[j] * scale + bias)) +
output_zeropoint; output_zeropoint;
output_ptr[j] = static_cast<uint8_t>( output_ptr[j] = static_cast<uint8_t>(
std::max<int32_t>(std::min<int32_t>(255, value), 0)); std::max<int32_t>(std::min<int32_t>(255, value), 0));
} }