blob: 3f8da5a92446d42a514d44a8b489fa56a35540ba (
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
|
/*
* Copyright (c) 2019 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 "nnfw_dev.h"
#include "flatbuffers/flexbuffers.h"
extern "C" void FillFromEval(nnfw_custom_kernel_params *params, char *userdata,
size_t userdata_size)
{
auto userdata_root = flexbuffers::GetRoot(reinterpret_cast<uint8_t *>(userdata), userdata_size);
auto attr_map = userdata_root.AsMap();
auto idx = attr_map["idx"].AsInt32();
auto val = attr_map["val"].AsFloat();
int32_t flat_size = 1;
for (int32_t i = 0; i < params->inputs[0].type.rank; ++i)
{
flat_size *= params->inputs[0].type.dims[i];
}
assert(idx <= flat_size);
auto output_flat = static_cast<float *>(params->outputs[0].allocation);
auto input_flat = static_cast<float *>(params->inputs[0].allocation);
for (int32_t i = 0; i < idx; ++i)
{
output_flat[i] = 0;
}
for (int32_t i = idx; i < flat_size; ++i)
{
output_flat[i] = val;
}
output_flat[idx] = input_flat[idx];
}
|
