FFmpeg
libavfilter
dnn
dnn_backend_native_layer_depth2space.h
Go to the documentation of this file.
1
/*
2
* Copyright (c) 2018 Sergey Lavrushkin
3
*
4
* This file is part of FFmpeg.
5
*
6
* FFmpeg is free software; you can redistribute it and/or
7
* modify it under the terms of the GNU Lesser General Public
8
* License as published by the Free Software Foundation; either
9
* version 2.1 of the License, or (at your option) any later version.
10
*
11
* FFmpeg is distributed in the hope that it will be useful,
12
* but WITHOUT ANY WARRANTY; without even the implied warranty of
13
* MERCHANTABILITY or FITNESS FOR A PARTICULAR PURPOSE. See the GNU
14
* Lesser General Public License for more details.
15
*
16
* You should have received a copy of the GNU Lesser General Public
17
* License along with FFmpeg; if not, write to the Free Software
18
* Foundation, Inc., 51 Franklin Street, Fifth Floor, Boston, MA 02110-1301 USA
19
*/
20
21
/**
22
* @file
23
* DNN inference functions interface for native backend.
24
*/
25
26
27
#ifndef AVFILTER_DNN_DNN_BACKEND_NATIVE_LAYER_DEPTH2SPACE_H
28
#define AVFILTER_DNN_DNN_BACKEND_NATIVE_LAYER_DEPTH2SPACE_H
29
30
#include "../dnn_interface.h"
31
#include "
libavformat/avio.h
"
32
33
typedef
struct
DepthToSpaceParams
{
34
int
block_size
;
35
}
DepthToSpaceParams
;
36
37
/**
38
* @brief Load the Depth to Space Layer.
39
*
40
* It assigns the depth to space layer with DepthToSpaceParams
41
* after parsing from the model file context.
42
*
43
* @param layer pointer to the DNN layer instance
44
* @param model_file_context pointer to model file context
45
* @param file_size model file size to check if data is read
46
* correctly from the model file
47
* @param operands_num operand count of the whole model to
48
* check if data is read correctly from the model file
49
* @return number of bytes read from the model file
50
* @retval 0 if an error occurs or out of memory
51
*/
52
int
ff_dnn_load_layer_depth2space
(
Layer
*layer,
AVIOContext
*model_file_context,
int
file_size,
int
operands_num);
53
54
/**
55
* @brief Execute the Depth to Space Layer.
56
*
57
* It rearranges the input data from depth into spatial
58
* form by applying Depth to Space transformation.
59
*
60
* @param operands all operands for the model
61
* @param input_operand_indexes input operand indexes for this layer
62
* @param output_operand_index output operand index for this layer
63
* @param parameters depth to space layer parameters
64
* @param ctx pointer to Native model context for logging
65
* @retval 0 if the execution succeeds
66
* @retval AVERROR(ENOMEM) if memory allocation fails
67
* @retval AVERROR(EINVAL) for invalid arguments
68
*/
69
int
ff_dnn_execute_layer_depth2space
(
DnnOperand
*operands,
const
int32_t
*input_operand_indexes,
70
int32_t
output_operand_index,
const
void
*parameters,
NativeContext
*
ctx
);
71
72
#endif
ff_dnn_load_layer_depth2space
int ff_dnn_load_layer_depth2space(Layer *layer, AVIOContext *model_file_context, int file_size, int operands_num)
Load the Depth to Space Layer.
Definition:
dnn_backend_native_layer_depth2space.c:29
ctx
AVFormatContext * ctx
Definition:
movenc.c:48
AVIOContext
Bytestream IO Context.
Definition:
avio.h:166
NativeContext
Definition:
dnn_backend_native.h:118
Layer
Definition:
dnn_backend_native.h:57
avio.h
DepthToSpaceParams
Definition:
dnn_backend_native_layer_depth2space.h:33
DepthToSpaceParams::block_size
int block_size
Definition:
dnn_backend_native_layer_depth2space.h:34
DnnOperand
Definition:
dnn_backend_native.h:69
int32_t
int32_t
Definition:
audioconvert.c:56
ff_dnn_execute_layer_depth2space
int ff_dnn_execute_layer_depth2space(DnnOperand *operands, const int32_t *input_operand_indexes, int32_t output_operand_index, const void *parameters, NativeContext *ctx)
Execute the Depth to Space Layer.
Definition:
dnn_backend_native_layer_depth2space.c:51
Generated on Tue Feb 28 2023 21:33:50 for FFmpeg by
1.8.17