ONE - On-device Neural Engine
Loading...
Searching...
No Matches
PALRound.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2023 Samsung Electronics Co., Ltd. All Rights Reserved
3
*
4
* Licensed under the Apache License, Version 2.0 (the "License");
5
* you may not use this file except in compliance with the License.
6
* You may obtain a copy of the License at
7
*
8
* http://www.apache.org/licenses/LICENSE-2.0
9
*
10
* Unless required by applicable law or agreed to in writing, software
11
* distributed under the License is distributed on an "AS IS" BASIS,
12
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
13
* See the License for the specific language governing permissions and
14
* limitations under the License.
15
*/
16
17
#ifndef LUCI_INTERPRETER_PAL_ROUND_COMMON_H
18
#define LUCI_INTERPRETER_PAL_ROUND_COMMON_H
19
20
namespace
luci_interpreter_pal
21
{
22
23
inline
float
RoundToNearest
(
float
value)
24
{
25
auto
floor_val = std::floor(value);
26
auto
diff = value - floor_val;
27
if
((diff < 0.5f) || ((diff == 0.5f) && (
static_cast<
int
>
(floor_val) % 2 == 0)))
28
{
29
return
floor_val;
30
}
31
else
32
{
33
return
floor_val + 1.0f;
34
}
35
}
36
37
inline
void
Round
(
const
int32_t flat_size,
const
float
*input_data,
float
*output_data)
38
{
39
for
(
int
i = 0; i < flat_size; ++i)
40
{
41
// Note that this implementation matches that of tensorFlow tf.round
42
// and corresponds to the bankers rounding method.
43
// cfenv (for fesetround) is not yet supported universally on Android, so
44
// using a work around.
45
output_data[i] =
RoundToNearest
(input_data[i]);
46
}
47
}
48
49
}
// namespace luci_interpreter_pal
50
51
#endif
// LUCI_INTERPRETER_PAL_ROUND_COMMON_H
luci_interpreter_pal
Definition
PALArgMax.h:23
luci_interpreter_pal::Round
void Round(const int32_t flat_size, const float *input_data, float *output_data)
Definition
PALRound.h:37
luci_interpreter_pal::RoundToNearest
float RoundToNearest(float value)
Definition
PALRound.h:23
onert-micro
luci-interpreter
pal
common
PALRound.h
Generated by
1.9.8