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
|
/*
* Copyright (c) 2020 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.
*/
#include "SelectLayer.h"
#include "OperationUtils.h"
#include <cker/operation/Select.h>
namespace onert
{
namespace backend
{
namespace cpu
{
namespace ops
{
SelectLayer::SelectLayer()
: _cond(nullptr), _input_true(nullptr), _input_false(nullptr), _output(nullptr)
{
// DO NOTHING
}
void SelectLayer::configure(const IPortableTensor *cond, const IPortableTensor *input_true,
const IPortableTensor *input_false, IPortableTensor *output)
{
_cond = cond;
_input_true = input_true;
_input_false = input_false;
_output = output;
}
void SelectLayer::run()
{
#define KERNEL_SELECT(type, op) \
nnfw::cker::op(getShape(_cond), getBuffer<uint8_t>(_cond), getShape(_input_true), \
getBuffer<type>(_input_true), getShape(_input_false), \
getBuffer<type>(_input_false), getShape(_output), getBuffer<type>(_output));
#define KERNEL_SWITCH(type, op) \
switch (type) \
{ \
break; \
case OperandType::FLOAT32: \
KERNEL_SELECT(float, op); \
break; \
default: \
throw std::runtime_error{"Select: unsupported data type"}; \
}
auto input_type = _input_true->data_type();
bool require_broadcast =
!HaveSameShapes(_input_true, _cond) || !HaveSameShapes(_input_false, _cond);
bool rank_one_select = ((_input_true->getShape().rank() == 1) && !require_broadcast);
if (rank_one_select)
{
KERNEL_SWITCH(input_type, RankOneSelect);
}
else if (require_broadcast)
{
KERNEL_SWITCH(input_type, BroadcastSelect4DSlow);
}
else
{
KERNEL_SWITCH(input_type, Select);
}
}
} // namespace ops
} // namespace cpu
} // namespace backend
} // namespace onert
|
