| /* Copyright 2018 The TensorFlow Authors. All Rights Reserved. | |
| Licensed under the Apache License, Version 2.0 (the "License"); | |
| you may not use this file except in compliance with the License. | |
| You may obtain a copy of the License at | |
| http://www.apache.org/licenses/LICENSE-2.0 | |
| Unless required by applicable law or agreed to in writing, software | |
| distributed under the License is distributed on an "AS IS" BASIS, | |
| WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. | |
| See the License for the specific language governing permissions and | |
| limitations under the License. | |
| ==============================================================================*/ | |
| namespace tflite { | |
| namespace reference_ops { | |
| inline float RoundToNearest(float value) { | |
| auto floor_val = std::floor(value); | |
| auto diff = value - floor_val; | |
| if ((diff < 0.5f) || | |
| ((diff == 0.5f) && (static_cast<int>(floor_val) % 2 == 0))) { | |
| return floor_val; | |
| } else { | |
| return floor_val = floor_val + 1.0f; | |
| } | |
| } | |
| inline void Round(const RuntimeShape& input_shape, const float* input_data, | |
| const RuntimeShape& output_shape, float* output_data) { | |
| const int flat_size = MatchingFlatSize(input_shape, output_shape); | |
| for (int i = 0; i < flat_size; ++i) { | |
| // Note that this implementation matches that of tensorFlow tf.round | |
| // and corresponds to the bankers rounding method. | |
| // cfenv (for fesetround) is not yet supported universally on Android, so | |
| // using a work around. | |
| output_data[i] = RoundToNearest(input_data[i]); | |
| } | |
| } | |
| } // namespace reference_ops | |
| } // namespace tflite | |