ONE - On-device Neural Engine
Loading...
Searching...
No Matches
EqualizePattern.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2024 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 __FME_APPLY_EQUALIZE_PATTERN_H__
18
#define __FME_APPLY_EQUALIZE_PATTERN_H__
19
20
#include <
luci/IR/CircleNode.h
>
21
22
namespace
fme_apply
23
{
24
25
struct
EqualizePattern
26
{
27
enum
Type
28
{
29
Invalid
,
// For debugging
30
ScaleOnly
,
31
ShiftOnly
,
32
ScaleShift
33
};
34
35
std::string
front
;
36
std::string
back
;
37
Type
type
=
Type::Invalid
;
38
std::vector<float>
act_scale
;
// activation channel scales
39
std::vector<float>
scale
;
// channel-wise scale
40
};
41
42
}
// namespace fme_apply
43
44
#endif
// __FME_APPLY_EQUALIZE_PATTERN_H__
CircleNode.h
fme_apply
Definition
EqualizePattern.h:23
fme_apply::EqualizePattern
Definition
EqualizePattern.h:26
fme_apply::EqualizePattern::Type
Type
Definition
EqualizePattern.h:28
fme_apply::EqualizePattern::ScaleOnly
@ ScaleOnly
Definition
EqualizePattern.h:30
fme_apply::EqualizePattern::Invalid
@ Invalid
Definition
EqualizePattern.h:29
fme_apply::EqualizePattern::ScaleShift
@ ScaleShift
Definition
EqualizePattern.h:32
fme_apply::EqualizePattern::ShiftOnly
@ ShiftOnly
Definition
EqualizePattern.h:31
fme_apply::EqualizePattern::front
std::string front
Definition
EqualizePattern.h:35
fme_apply::EqualizePattern::scale
std::vector< float > scale
Definition
EqualizePattern.h:39
fme_apply::EqualizePattern::act_scale
std::vector< float > act_scale
Definition
EqualizePattern.h:38
fme_apply::EqualizePattern::type
Type type
Definition
EqualizePattern.h:37
fme_apply::EqualizePattern::back
std::string back
Definition
EqualizePattern.h:36
compiler
fme-apply
src
EqualizePattern.h
Generated by
1.9.8