diff options
author | Guo, Yejun <yejun.guo@intel.com> | 2019-10-09 22:08:18 +0800 |
---|---|---|
committer | Pedro Arthur <bygrandao@gmail.com> | 2019-10-15 18:56:54 -0300 |
commit | 2558e62713ebc5f3ea22c1a28d8e9cf3249badaf (patch) | |
tree | c8b4dfe51ec30faa41f7327f15000d49c3f05c72 /libavfilter/dnn/dnn_backend_native_layer_maximum.c | |
parent | 3fd5ac7e92049b4f31026acdb53a762289f71448 (diff) | |
download | ffmpeg-streaming-2558e62713ebc5f3ea22c1a28d8e9cf3249badaf.zip ffmpeg-streaming-2558e62713ebc5f3ea22c1a28d8e9cf3249badaf.tar.gz |
avfilter/dnn: unify the layer load function in native mode
Signed-off-by: Guo, Yejun <yejun.guo@intel.com>
Signed-off-by: Pedro Arthur <bygrandao@gmail.com>
Diffstat (limited to 'libavfilter/dnn/dnn_backend_native_layer_maximum.c')
-rw-r--r-- | libavfilter/dnn/dnn_backend_native_layer_maximum.c | 18 |
1 files changed, 18 insertions, 0 deletions
diff --git a/libavfilter/dnn/dnn_backend_native_layer_maximum.c b/libavfilter/dnn/dnn_backend_native_layer_maximum.c index 6add170..19f0e8d 100644 --- a/libavfilter/dnn/dnn_backend_native_layer_maximum.c +++ b/libavfilter/dnn/dnn_backend_native_layer_maximum.c @@ -27,6 +27,24 @@ #include "libavutil/avassert.h" #include "dnn_backend_native_layer_maximum.h" +int dnn_load_layer_maximum(Layer *layer, AVIOContext *model_file_context, int file_size) +{ + DnnLayerMaximumParams *params; + int dnn_size = 0; + params = av_malloc(sizeof(*params)); + if (!params) + return 0; + + params->val.u32 = avio_rl32(model_file_context); + dnn_size += 4; + layer->params = params; + layer->input_operand_indexes[0] = (int32_t)avio_rl32(model_file_context); + layer->output_operand_index = (int32_t)avio_rl32(model_file_context); + dnn_size += 8; + + return dnn_size; +} + int dnn_execute_layer_maximum(DnnOperand *operands, const int32_t *input_operand_indexes, int32_t output_operand_index, const void *parameters) { |