dequantize.h

00001 /* Copyright 2019 The TensorFlow Authors. All Rights Reserved.
00002 
00003 Licensed under the Apache License, Version 2.0 (the "License");
00004 you may not use this file except in compliance with the License.
00005 You may obtain a copy of the License at
00006 
00007     http://www.apache.org/licenses/LICENSE-2.0
00008 
00009 Unless required by applicable law or agreed to in writing, software
00010 distributed under the License is distributed on an "AS IS" BASIS,
00011 WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
00012 See the License for the specific language governing permissions and
00013 limitations under the License.
00014 ==============================================================================*/
00015 #ifndef TENSORFLOW_LITE_KERNELS_INTERNAL_REFERENCE_DEQUANTIZE_H_
00016 #define TENSORFLOW_LITE_KERNELS_INTERNAL_REFERENCE_DEQUANTIZE_H_
00017 
00018 #include "tensorflow/lite/kernels/internal/common.h"
00019 #include "tensorflow/lite/kernels/internal/types.h"
00020 
00021 namespace tflite {
00022 
00023 namespace reference_ops {
00024 
00025 template <typename T>
00026 inline void Dequantize(const tflite::DequantizationParams& op_params,
00027                        const RuntimeShape& input_shape, const T* input_data,
00028                        const RuntimeShape& output_shape, float* output_data) {
00029   int32 zero_point = op_params.zero_point;
00030   const double scale = op_params.scale;
00031   const int flat_size = MatchingFlatSize(input_shape, output_shape);
00032 
00033   for (int i = 0; i < flat_size; i++) {
00034     const int32 val = input_data[i];
00035     const float result = static_cast<float>(scale * (val - zero_point));
00036     output_data[i] = result;
00037   }
00038 }
00039 
00040 }  // namespace reference_ops
00041 
00042 }  // namespace tflite
00043 #endif  // TENSORFLOW_LITE_KERNELS_INTERNAL_REFERENCE_DEQUANTIZE_H_