ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Tanh.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2020 Samsung Electronics Co., Ltd. All Rights Reserved
3
* Copyright 2018 The TensorFlow Authors. All Rights Reserved.
4
*
5
* Licensed under the Apache License, Version 2.0 (the "License");
6
* you may not use this file except in compliance with the License.
7
* You may obtain a copy of the License at
8
*
9
* http://www.apache.org/licenses/LICENSE-2.0
10
*
11
* Unless required by applicable law or agreed to in writing, software
12
* distributed under the License is distributed on an "AS IS" BASIS,
13
* WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
14
* See the License for the specific language governing permissions and
15
* limitations under the License.
16
*/
17
18
#ifndef __NNFW_CKER_TANH_H__
19
#define __NNFW_CKER_TANH_H__
20
21
#include "
cker/eigen/Utils.h
"
22
#include "
cker/Shape.h
"
23
#include "
cker/Types.h
"
24
#include <Eigen/Core>
25
26
namespace
nnfw
27
{
28
namespace
cker
29
{
30
31
inline
void
Tanh
(
const
Shape
&input_shape,
const
float
*input_data,
const
Shape
&
output_shape
,
32
float
*output_data)
33
{
34
auto
input_map =
MapAsVector
(input_data, input_shape);
35
auto
output_map =
MapAsVector
(output_data,
output_shape
);
36
output_map.array() = input_map.array().tanh();
37
}
38
39
}
// namespace cker
40
}
// namespace nnfw
41
42
#endif
// __NNFW_CKER_TANH_H__
nnfw::cker::Shape
Definition
Shape.h:32
Shape.h
Types.h
Utils.h
output_shape
const luci_interpreter::RuntimeShape output_shape
Definition
PALComparisons.h:32
nnfw::cker::Tanh
void Tanh(const Shape &input_shape, const float *input_data, const Shape &output_shape, float *output_data)
Definition
Tanh.h:31
nnfw::cker::MapAsVector
VectorMap< Scalar > MapAsVector(Scalar *data, const Shape &shape)
Definition
Utils.h:43
nnfw
Definition
topk_v2.h:30
compute
cker
include
cker
operation
Tanh.h
Generated by
1.9.8