virtualenv/FFmpeg
1
0
Fork 0
mirror of https://github.com/FFmpeg/FFmpeg.git synced 2024-12-23 12:43:46 +02:00
Code Issues Releases Activity

lavfi/dnn_backend_tf: Error Handling for tf_create_inference_request

Browse Source 
This commit includes the check for the case when the newly created
TFInferRequest is NULL.

Signed-off-by: Shubhanshu Saxena <shubhanshu.e01@gmail.com>
This commit is contained in:
Shubhanshu Saxena 2021-08-08 16:25:38 +05:30 committed by Guo Yejun
parent 009b2e5b5e
commit 371e5672f3
1 changed files with 3 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

3
libavfilter/dnn/dnn_backend_tf.c
View File

@ -135,6 +135,9 @@ static void tf_free_request(TFInferRequest *request)
static TFInferRequest *tf_create_inference_request(void) static TFInferRequest *tf_create_inference_request(void)
{ {
TFInferRequest *infer_request = av_malloc(sizeof(TFInferRequest)); TFInferRequest *infer_request = av_malloc(sizeof(TFInferRequest));
if (!infer_request) {
return NULL;
}
infer_request->tf_outputs = NULL; infer_request->tf_outputs = NULL;
infer_request->tf_input = NULL; infer_request->tf_input = NULL;
infer_request->input_tensor = NULL; infer_request->input_tensor = NULL;