blob: f51f0402a52c0168d9c2d665933e5f45acbae907 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
110
111
112
113
114
115
116
117
118
119
120
121
122
123
124
125
126
127
128
129
130
131
|
/*
* Copyright (c) 2018 Samsung Electronics Co., Ltd. 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.
*/
/**
* @file Lstm.h
* @ingroup COM_AI_RUNTIME
* @brief This file defines internal::tflite::op::LSTM::Param struct
* and internal::tflite::op::LSTM::Node class
*/
#ifndef __INTERNAL_OP_LSTM_H__
#define __INTERNAL_OP_LSTM_H__
#include "internal/op/Node.h"
#include <cstdint>
namespace internal
{
namespace tflite
{
namespace op
{
namespace LSTM
{
/**
* @brief Struct to have indexes for operation parameter
*/
struct Param
{
int32_t scratch_buffer_index; /**< Index of scartch buffer */
int32_t output_state_out_index; /**< Index of output state out */
int32_t cell_state_out_index; /**< Index of cell state out */
int32_t output_index; /**< Index of output */
int32_t input_index; /**< Index of input */
int32_t input_to_input_weights_index; /**< Index of input to input weights */
int32_t input_to_forget_weights_index; /**< Index of input to forget weights */
int32_t input_to_cell_weights_index; /**< Index of input to cell weights */
int32_t input_to_output_weights_index; /**< Index of input to output weights */
int32_t recurrent_to_input_weights_index; /**< Index of recurrent to input weights */
int32_t recurrent_to_forget_weights_index; /**< Index of recurrent to forget weights */
int32_t recurrent_to_cell_weights_index; /**< Index of recurrent to cell weights */
int32_t recurrent_to_output_weights_index; /**< Index of recurrent to output weights */
int32_t cell_to_input_weights_index; /**< Index of cell to input weights */
int32_t cell_to_forget_weights_index; /**< Index of cell to forget weights */
int32_t cell_to_output_weights_index; /**< Index of cell to output weights */
int32_t input_gate_bias_index; /**< Index of input gate bias */
int32_t forget_gate_bias_index; /**< Index of forget gate bias */
int32_t cell_bias_index; /**< Index of cell bias */
int32_t output_gate_bias_index; /**< Index of output gate bias */
int32_t projection_weights_index; /**< Index of projection weights */
int32_t projection_bias_index; /**< Index of projection bias */
int32_t output_state_in_index; /**< Index of output state in */
int32_t cell_state_in_index; /**< Index of cell state in */
int32_t activation_index; /**< Index of activation */
int32_t cell_threshold_index; /**< Index of cell threshold */
int32_t projection_threshold_index; /**< Index of projection threshold */
/**
* @brief Construct as default
*/
Param() = default;
/**
* @brief Construct a new Param object with params
* @param[in] inputCount Count of inputs
* @param[in] inputs Pointer of inputs
* @param[in] outputCount Count of outputs
* @param[in] outputs Pointer of outputs
*/
Param(uint32_t inputCount, const uint32_t *inputs, uint32_t outputCount, const uint32_t *outputs);
};
/**
* @brief Class to represent an operation of data structure
*/
class Node final : public op::Node
{
public:
/**
* @brief Construct a new Node object with param
* @param[in] param Param object that makes up a Node
*/
Node(const Param ¶m) : _param(param)
{
// DO NOTHING
}
public:
/**
* @brief Destruct as default
*/
virtual ~Node() = default;
public:
/**
* @brief Get a reference of Param object
* @return Reference of Param object
*/
const Param ¶m(void) const { return _param; }
public:
/**
* @brief Visit this Node by NodeVisitor
* @return N/A
*/
void accept(NodeVisitor &&) const override;
private:
const Param _param;
};
} // namespace LSTM
} // namespace op
} // namespace tflite
} // namespace internal
#endif // __INTERNAL_OP_LSTM_H__
|
