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"

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 (const DNNModel *model, const char *input_name, AVFrame *in_frame, const char **output_names, uint32_t nb_output, AVFrame *out_frame, int do_ioproc)
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, const char *input_name, AVFrame *in_frame, const char **output_names, uint32_t nb_output, AVFrame *out_frame)
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 32 of file dnn_backend_native.c.

FLAGS

#define FLAGS   AV_OPT_FLAG_FILTERING_PARAM

Definition at line 33 of file dnn_backend_native.c.

DNN_NATIVE_MAGIC

#define DNN_NATIVE_MAGIC   "FFMPEGDNNNATIVE"

Function Documentation

execute_model_native()

static DNNReturnType execute_model_native ( const DNNModel *  model, const char *  input_name, AVFrame *  in_frame, const char **  output_names, uint32_t  nb_output, AVFrame *  out_frame, int  do_ioproc  )

static

Definition at line 261 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 51 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 77 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 115 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,
		const char *	input_name,
		AVFrame *	in_frame,
		const char **	output_names,
		uint32_t	nb_output,
		AVFrame *	out_frame
	)

Definition at line 375 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 394 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 403 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 415 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 },
    { NULL },
}

Definition at line 34 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,
    .version    = LIBAVUTIL_VERSION_INT,
    .category   = AV_CLASS_CATEGORY_FILTER,
}

Definition at line 39 of file dnn_backend_native.c.

Referenced by ff_dnn_load_model_native().