blob: ecf0597cfe8b1b09646997c09ed6b2d5fb2de414 (
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
|
/*
* 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.
*/
#ifndef __NNFW_DEV_H__
#define __NNFW_DEV_H__
#include "nnfw.h"
// Used for custom kernel development
/*
* operand type, used only for custom operations
*/
typedef struct
{
nnfw_tensorinfo type;
void *allocation;
} nnfw_operand;
/*
* Used as input to custom operation eval function
*/
typedef struct
{
size_t ninputs;
nnfw_operand *inputs;
size_t noutputs;
nnfw_operand *outputs;
} nnfw_custom_kernel_params;
/*
* Custom kernel evaluation function
*
* param[in] params custom operation parameters
* param[in] userdata pointer to user-specified buffer( kernel instance specific )
*/
typedef void (*nnfw_custom_eval)(nnfw_custom_kernel_params *params, char *userdata,
size_t userdata_size);
/*
* custom operation registration info
*/
typedef struct
{
nnfw_custom_eval eval_function;
} custom_kernel_registration_info;
NNFW_STATUS nnfw_register_custom_op_info(nnfw_session *session, const char *id,
custom_kernel_registration_info *info);
#endif // __NNFW_DEV_H__
|
