diff options
author | Ting Fu <ting.fu@intel.com> | 2020-08-25 06:47:49 +0300 |
---|---|---|
committer | Guo, Yejun <yejun.guo@intel.com> | 2020-08-25 08:03:46 +0300 |
commit | 230cf9d1854b171727d1b10c2a2fbe0df9d1d489 (patch) | |
tree | 1daaa8ef9118e1b4eef9801c04e621944b57d4ee /libavfilter/dnn/dnn_backend_native.c | |
parent | 0f7a99e37ae52f9ecdc4c81195c14b03f5be3dfd (diff) |
dnn/native: unify error return to DNN_ERROR
Unify all error return as DNN_ERROR, in order to cease model executing
when return error in ff_dnn_execute_model_native layer_func.pf_exec
Signed-off-by: Ting Fu <ting.fu@intel.com>
Diffstat (limited to 'libavfilter/dnn/dnn_backend_native.c')
-rw-r--r-- | libavfilter/dnn/dnn_backend_native.c | 10 |
1 files changed, 6 insertions, 4 deletions
diff --git a/libavfilter/dnn/dnn_backend_native.c b/libavfilter/dnn/dnn_backend_native.c index 65a56704d3..436ce938da 100644 --- a/libavfilter/dnn/dnn_backend_native.c +++ b/libavfilter/dnn/dnn_backend_native.c @@ -246,10 +246,12 @@ DNNReturnType ff_dnn_execute_model_native(const DNNModel *model, DNNData *output for (layer = 0; layer < native_model->layers_num; ++layer){ DNNLayerType layer_type = native_model->layers[layer].type; - layer_funcs[layer_type].pf_exec(native_model->operands, - native_model->layers[layer].input_operand_indexes, - native_model->layers[layer].output_operand_index, - native_model->layers[layer].params); + if (layer_funcs[layer_type].pf_exec(native_model->operands, + native_model->layers[layer].input_operand_indexes, + native_model->layers[layer].output_operand_index, + native_model->layers[layer].params) == DNN_ERROR) { + return DNN_ERROR; + } } for (uint32_t i = 0; i < nb_output; ++i) { |