[C-Api] fill dim (default 1)
authorJaeyun Jung <jy1210.jung@samsung.com>
Fri, 7 Apr 2023 07:50:33 +0000 (16:50 +0900)
committerSangjung Woo <again4you@gmail.com>
Tue, 11 Apr 2023 08:16:10 +0000 (17:16 +0900)
When calling get-dimension(), fill remained dim - default dim is 1.

Signed-off-by: Jaeyun Jung <jy1210.jung@samsung.com>
c/src/ml-api-common.c

index c78f96c..b09e750 100644 (file)
@@ -551,6 +551,10 @@ ml_tensors_info_get_tensor_dimension (ml_tensors_info_h info,
     dimension[i] = tensors_info->info[index].dimension[i];
   }
 
+  /* Fill remained dim (default value 1) */
+  for (i = valid_rank; i < ML_TENSOR_RANK_LIMIT; i++)
+    dimension[i] = 1;
+
   G_UNLOCK_UNLESS_NOLOCK (*tensors_info);
   return ML_ERROR_NONE;
 }