virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-11-26 19:01:44 +02:00
Code Issues Releases Activity

dnn_backend_native_layer_mathbinary: change to function pointer

Browse Source 
Signed-off-by: Mingyu Yin <mingyu.yin@intel.com>
This commit is contained in:
Mingyu Yin 2020-08-23 23:12:12 +08:00 committed by Guo, Yejun
parent c8716b5029
commit 37ef1acedb
1 changed files with 73 additions and 70 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

143
libavfilter/dnn/dnn_backend_native_layer_mathbinary.c
View File

@ -27,6 +27,74 @@
#include "libavutil/avassert.h"
#include "dnn_backend_native_layer_mathbinary.h"
typedef float (*FunType)(float src0, float src1);
FunType pfun;
static float sub(float src0, float src1)
{
return src0 - src1;
}
static float add(float src0, float src1)
{
return src0 + src1;
}
static float mul(float src0, float src1)
{
return src0 * src1;
}
static float realdiv(float src0, float src1)
{
return src0 / src1;
}
static float minimum(float src0, float src1)
{
return FFMIN(src0, src1);
}
static void math_binary_commutative(FunType pfun, const DnnLayerMathBinaryParams *params, const DnnOperand *input, DnnOperand *output, DnnOperand *operands, const int32_t *input_operand_indexes)
{
int dims_count;
const float *src;
float *dst;
dims_count = calculate_operand_dims_count(output);
src = input->data;
dst = output->data;
if (params->input0_broadcast || params->input1_broadcast) {
for (int i = 0; i < dims_count; ++i) {
dst[i] = pfun(params->v, src[i]);
}
} else {
const DnnOperand *input1 = &operands[input_operand_indexes[1]];
const float *src1 = input1->data;
for (int i = 0; i < dims_count; ++i) {
dst[i] = pfun(src[i], src1[i]);
}
}
}
static void math_binary_not_commutative(FunType pfun, const DnnLayerMathBinaryParams *params, const DnnOperand *input, DnnOperand *output, DnnOperand *operands, const int32_t *input_operand_indexes)
{
int dims_count;
const float *src;
float *dst;
dims_count = calculate_operand_dims_count(output);
src = input->data;
dst = output->data;
if (params->input0_broadcast) {
for (int i = 0; i < dims_count; ++i) {
dst[i] = pfun(params->v, src[i]);
}
} else if (params->input1_broadcast) {
for (int i = 0; i < dims_count; ++i) {
dst[i] = pfun(src[i], params->v);
}
} else {
const DnnOperand *input1 = &operands[input_operand_indexes[1]];
const float *src1 = input1->data;
for (int i = 0; i < dims_count; ++i) {
dst[i] = pfun(src[i], src1[i]);
}
}
}
int dnn_load_layer_math_binary(Layer *layer, AVIOContext *model_file_context, int file_size, int operands_num)
{
DnnLayerMathBinaryParams *params;
@ -82,9 +150,6 @@ int dnn_execute_layer_math_binary(DnnOperand *operands, const int32_t *input_ope
const DnnOperand *input = &operands[input_operand_indexes[0]];
DnnOperand *output = &operands[output_operand_index];
const DnnLayerMathBinaryParams *params = (const DnnLayerMathBinaryParams *)parameters;
int dims_count;
const float *src;
float *dst;
for (int i = 0; i < 4; ++i)
output->dims[i] = input->dims[i];
@ -97,83 +162,21 @@ int dnn_execute_layer_math_binary(DnnOperand *operands, const int32_t *input_ope
if (!output->data)
return DNN_ERROR;
dims_count = calculate_operand_dims_count(output);
src = input->data;
dst = output->data;
switch (params->bin_op) {
case DMBO_SUB:
if (params->input0_broadcast) {
for (int i = 0; i < dims_count; ++i) {
dst[i] = params->v - src[i];
}
} else if (params->input1_broadcast) {
for (int i = 0; i < dims_count; ++i) {
dst[i] = src[i] - params->v;
}
} else {
const DnnOperand *input1 = &operands[input_operand_indexes[1]];
const float *src1 = input1->data;
for (int i = 0; i < dims_count; ++i) {
dst[i] = src[i] - src1[i];
}
}
math_binary_not_commutative(sub, params, input, output, operands, input_operand_indexes);
return 0;
case DMBO_ADD:
if (params->input0_broadcast || params->input1_broadcast) {
for (int i = 0; i < dims_count; ++i) {
dst[i] = params->v + src[i];
}
} else {
const DnnOperand *input1 = &operands[input_operand_indexes[1]];
const float *src1 = input1->data;
for (int i = 0; i < dims_count; ++i) {
dst[i] = src[i] + src1[i];
}
}
math_binary_commutative(add, params, input, output, operands, input_operand_indexes);
return 0;
case DMBO_MUL:
if (params->input0_broadcast || params->input1_broadcast) {
for (int i = 0; i < dims_count; ++i) {
dst[i] = params->v * src[i];
}
} else {
const DnnOperand *input1 = &operands[input_operand_indexes[1]];
const float *src1 = input1->data;
for (int i = 0; i < dims_count; ++i) {
dst[i] = src[i] * src1[i];
}
}
math_binary_commutative(mul, params, input, output, operands, input_operand_indexes);
return 0;
case DMBO_REALDIV:
if (params->input0_broadcast) {
for (int i = 0; i < dims_count; ++i) {
dst[i] = params->v / src[i];
}
} else if (params->input1_broadcast) {
for (int i = 0; i < dims_count; ++i) {
dst[i] = src[i] / params->v;
}
} else {
const DnnOperand *input1 = &operands[input_operand_indexes[1]];
const float *src1 = input1->data;
for (int i = 0; i < dims_count; ++i) {
dst[i] = src[i] / src1[i];
}
}
math_binary_not_commutative(realdiv, params, input, output, operands, input_operand_indexes);
return 0;
case DMBO_MINIMUM:
if (params->input0_broadcast || params->input1_broadcast) {
for (int i = 0; i < dims_count; ++i) {
dst[i] = FFMIN(params->v, src[i]);
}
} else {
const DnnOperand *input1 = &operands[input_operand_indexes[1]];
const float *src1 = input1->data;
for (int i = 0; i < dims_count; ++i) {
dst[i] = FFMIN(src[i], src1[i]);
}
}
math_binary_commutative(minimum, params, input, output, operands, input_operand_indexes);
return 0;
default:
return -1;