DNN tensorflow backend implementation. More...

#include "dnn_backend_tf.h"
#include "dnn_backend_native.h"
#include "libavformat/avio.h"
#include <tensorflow/c/c_api.h>

Functions
static void	free_buffer (void *data, size_t length)

static TF_Buffer *	read_graph (const char *model_filename)

static DNNReturnType	set_input_output_tf (void model, DNNData input, DNNData *output)

static DNNReturnType	load_tf_model (TFModel tf_model, const char model_filename)

static DNNReturnType	add_conv_layer (TFModel tf_model, TF_Operation transpose_op, TF_Operation *cur_op, ConvolutionalParams params, const int layer)

static DNNReturnType	add_depth_to_space_layer (TFModel tf_model, TF_Operation cur_op, DepthToSpaceParams params, const int layer)

static int	calculate_pad (const ConvolutionalNetwork *conv_network)

static DNNReturnType	add_pad_op (TFModel tf_model, TF_Operation *cur_op, const int32_t pad)

static DNNReturnType	load_native_model (TFModel tf_model, const char model_filename)

DNNModel *	ff_dnn_load_model_tf (const char *model_filename)

DNNReturnType	ff_dnn_execute_model_tf (const DNNModel *model)

void	ff_dnn_free_model_tf (DNNModel **model)

Detailed Description

DNN tensorflow backend implementation.

Definition in file dnn_backend_tf.c.

Macro Definition Documentation

#define NAME_BUFFER_SIZE 256

Definition at line 182 of file dnn_backend_tf.c.

#define CLEANUP_ON_ERROR ( tf_model )

Value:

{ \
        TF_DeleteGraph(tf_model->graph); \
        TF_DeleteStatus(tf_model->status); \
        return DNN_ERROR; \
    }

static

Definition at line 41 of file dnn_backend_tf.c.

Referenced by read_graph().

static TF_Buffer* read_graph ( const char * model_filename )

static

Definition at line 46 of file dnn_backend_tf.c.

static DNNReturnType set_input_output_tf	(	void *	model,
		DNNData *	input,
		DNNData *	output
	)

static

Definition at line 79 of file dnn_backend_tf.c.

static DNNReturnType load_tf_model	(	TFModel *	tf_model,
		const char *	model_filename
	)

static

Definition at line 158 of file dnn_backend_tf.c.

static DNNReturnType add_conv_layer	(	TFModel *	tf_model,
		TF_Operation *	transpose_op,
		TF_Operation **	cur_op,
		ConvolutionalParams *	params,
		const int	layer
	)