[single] Allow tvn model when using VD_AIFW
authoryelini-jeong <yelini.jeong@samsung.com>
Mon, 18 Apr 2022 09:55:54 +0000 (18:55 +0900)
committerjaeyun-jung <39614140+jaeyun-jung@users.noreply.github.com>
Mon, 18 Apr 2022 23:13:55 +0000 (08:13 +0900)
This patch allows tvn model when using VD_AIFW framework.

Signed-off-by: yelini-jeong <yelini.jeong@samsung.com>
c/src/ml-api-inference-single.c

index d7e2e8a..1cf2944 100644 (file)
@@ -1879,6 +1879,7 @@ _ml_validate_model_file (const char *const *model,
     case ML_NNFW_TYPE_VD_AIFW:
       if (!g_str_equal (file_ext[0], ".nb") &&
           !g_str_equal (file_ext[0], ".ncp") &&
+          !g_str_equal (file_ext[0], ".tvn") &&
           !g_str_equal (file_ext[0], ".bin")) {
         status = ML_ERROR_INVALID_PARAMETER;
       }