FFmpeg
Macros | Functions | Variables
dnn_backend_native.c File Reference
#include "dnn_backend_native.h"
#include "libavutil/avassert.h"
#include "dnn_backend_native_layer_conv2d.h"
#include "dnn_backend_native_layers.h"
#include "dnn_io_proc.h"
#include "dnn_backend_common.h"

Go to the source code of this file.

Macros

#define  OFFSET(x)   offsetof(NativeContext, x)
 
#define  FLAGS   AV_OPT_FLAG_FILTERING_PARAM
 
#define  DNN_NATIVE_MAGIC   "FFMPEGDNNNATIVE"
 

Functions

static DNNReturnType  execute_model_native (Queue *lltask_queue)
 
static DNNReturnType  extract_lltask_from_task (TaskItem *task, Queue *lltask_queue)
 
static DNNReturnType  get_input_native (void *model, DNNData *input, const char *input_name)
 
static DNNReturnType  get_output_native (void *model, const char *input_name, int input_width, int input_height, const char *output_name, int *output_width, int *output_height)
 
DNNModel *  ff_dnn_load_model_native (const char *model_filename, DNNFunctionType func_type, const char *options, AVFilterContext *filter_ctx)
 
DNNReturnType  ff_dnn_execute_model_native (const DNNModel *model, DNNExecBaseParams *exec_params)
 
DNNReturnType  ff_dnn_flush_native (const DNNModel *model)
 
DNNAsyncStatusType  ff_dnn_get_result_native (const DNNModel *model, AVFrame **in, AVFrame **out)
 
int32_t  ff_calculate_operand_dims_count (const DnnOperand *oprd)
 
int32_t  ff_calculate_operand_data_length (const DnnOperand *oprd)
 
void  ff_dnn_free_model_native (DNNModel **model)
 

Variables

static const AVOption  dnn_native_options []
 
static const AVClass  dnn_native_class
 

Detailed Description

DNN native backend implementation.

Definition in file dnn_backend_native.c.

Macro Definition Documentation

◆ OFFSET

#define OFFSET (   x )    offsetof(NativeContext, x)

Definition at line 33 of file dnn_backend_native.c.

◆ FLAGS

#define FLAGS   AV_OPT_FLAG_FILTERING_PARAM

Definition at line 34 of file dnn_backend_native.c.

◆ DNN_NATIVE_MAGIC

#define DNN_NATIVE_MAGIC   "FFMPEGDNNNATIVE"

Function Documentation

◆ execute_model_native()

static DNNReturnType execute_model_native ( Queue *  lltask_queue )
static

Definition at line 300 of file dnn_backend_native.c.

Referenced by ff_dnn_execute_model_native(), ff_dnn_flush_native(), and get_output_native().

◆ extract_lltask_from_task()

static DNNReturnType extract_lltask_from_task ( TaskItem *  task,
Queue *  lltask_queue 
)
static

Definition at line 51 of file dnn_backend_native.c.

Referenced by ff_dnn_execute_model_native(), and get_output_native().

◆ get_input_native()

static DNNReturnType get_input_native ( void *  model,
DNNData *  input,
const char *  input_name 
)
static

Definition at line 73 of file dnn_backend_native.c.

Referenced by ff_dnn_load_model_native().

◆ get_output_native()

static DNNReturnType get_output_native ( void *  model,
const char *  input_name,
int  input_width,
int  input_height,
const char *  output_name,
int *  output_width,
int *  output_height 
)
static

Definition at line 99 of file dnn_backend_native.c.

Referenced by ff_dnn_load_model_native().

◆ ff_dnn_load_model_native()

DNNModel* ff_dnn_load_model_native ( const char *  model_filename,
DNNFunctionType  func_type,
const char *  options,
AVFilterContext *  filter_ctx 
)

check file header with string and version

Definition at line 139 of file dnn_backend_native.c.

Referenced by ff_get_dnn_module(), and load_native_model().

◆ ff_dnn_execute_model_native()

DNNReturnType ff_dnn_execute_model_native ( const DNNModel *  model,
DNNExecBaseParams *  exec_params 
)

Definition at line 435 of file dnn_backend_native.c.

Referenced by ff_get_dnn_module().

◆ ff_dnn_flush_native()

DNNReturnType ff_dnn_flush_native ( const DNNModel *  model )

Definition at line 470 of file dnn_backend_native.c.

Referenced by ff_get_dnn_module().

◆ ff_dnn_get_result_native()

DNNAsyncStatusType ff_dnn_get_result_native ( const DNNModel *  model,
AVFrame **  in,
AVFrame **  out 
)

Definition at line 484 of file dnn_backend_native.c.

Referenced by ff_get_dnn_module().

◆ ff_calculate_operand_dims_count()

int32_t ff_calculate_operand_dims_count ( const DnnOperand *  oprd )

Definition at line 490 of file dnn_backend_native.c.

Referenced by ff_dnn_execute_layer_math_unary(), ff_dnn_execute_layer_maximum(), math_binary_commutative(), and math_binary_not_commutative().

◆ ff_calculate_operand_data_length()

int32_t ff_calculate_operand_data_length ( const DnnOperand *  oprd )

Definition at line 499 of file dnn_backend_native.c.

Referenced by execute_model_native(), ff_dnn_execute_layer_avg_pool(), ff_dnn_execute_layer_conv2d(), ff_dnn_execute_layer_dense(), ff_dnn_execute_layer_depth2space(), ff_dnn_execute_layer_math_binary(), ff_dnn_execute_layer_math_unary(), ff_dnn_execute_layer_maximum(), and ff_dnn_execute_layer_pad().

◆ ff_dnn_free_model_native()

void ff_dnn_free_model_native ( DNNModel **  model )

Definition at line 511 of file dnn_backend_native.c.

Referenced by ff_dnn_load_model_native(), ff_get_dnn_module(), and load_native_model().

Variable Documentation

◆ dnn_native_options

const AVOption dnn_native_options[]
static
Initial value:
= {
{ "conv2d_threads", "threads num for conv2d layer", OFFSET(options.conv2d_threads), AV_OPT_TYPE_INT, { .i64 = 0 }, INT_MIN, INT_MAX, FLAGS },
{ "async", "use DNN async inference", OFFSET(options.async), AV_OPT_TYPE_BOOL, { .i64 = 0 }, 0, 1, FLAGS },
{ NULL },
}

Definition at line 35 of file dnn_backend_native.c.

◆ dnn_native_class

const AVClass dnn_native_class
static
Initial value:
= {
.class_name = "dnn_native",
.item_name = av_default_item_name,
.option = dnn_native_options,
}

Definition at line 41 of file dnn_backend_native.c.

Referenced by ff_dnn_load_model_native().

OFFSET
#define OFFSET(x)
Definition: dnn_backend_native.c:33
LIBAVUTIL_VERSION_INT
#define LIBAVUTIL_VERSION_INT
Definition: version.h:85
NULL
#define NULL
Definition: coverity.c:32
av_default_item_name
const char * av_default_item_name(void *ptr)
Return the context name.
Definition: log.c:235
dnn_native_options
static const AVOption dnn_native_options[]
Definition: dnn_backend_native.c:35
AV_CLASS_CATEGORY_FILTER
@ AV_CLASS_CATEGORY_FILTER
Definition: log.h:36
options
const OptionDef options[]
AV_OPT_TYPE_INT
@ AV_OPT_TYPE_INT
Definition: opt.h:224
FLAGS
#define FLAGS
Definition: dnn_backend_native.c:34
AV_OPT_TYPE_BOOL
@ AV_OPT_TYPE_BOOL
Definition: opt.h:241
Generated on Wed Aug 24 2022 21:38:45 for FFmpeg by   doxygen 1.8.17

AltStyle によって変換されたページ (->オリジナル) /