[Common] condition to compare tensor info
authorJaeyun <jy1210.jung@samsung.com>
Thu, 28 Nov 2019 10:57:55 +0000 (19:57 +0900)
committerMyungJoo Ham <myungjoo.ham@samsung.com>
Tue, 3 Dec 2019 09:37:05 +0000 (18:37 +0900)
1. validate tensor info when comparing the structure.
2. fix invalid doxygen tag.

Signed-off-by: Jaeyun Jung <jy1210.jung@samsung.com>
gst/nnstreamer/nnstreamer_plugin_api.h
gst/nnstreamer/tensor_common.c

index 987dc74..682df03 100644 (file)
@@ -64,7 +64,7 @@ gst_tensor_info_validate (const GstTensorInfo *info);
 
 /**
  * @brief Compare tensor info
- * @param TRUE if equal
+ * @return TRUE if equal, FALSE if given tensor infos are invalid or not equal.
  */
 extern gboolean
 gst_tensor_info_is_equal (const GstTensorInfo * i1, const GstTensorInfo * i2);
@@ -172,7 +172,7 @@ gst_tensors_info_validate (const GstTensorsInfo *info);
 
 /**
  * @brief Compare tensors info
- * @param TRUE if equal
+ * @return TRUE if equal, FALSE if given tensor infos are invalid or not equal.
  */
 extern gboolean
 gst_tensors_info_is_equal (const GstTensorsInfo * i1, const GstTensorsInfo * i2);
index ae1b8f4..a2d1467 100644 (file)
@@ -167,15 +167,17 @@ gst_tensor_info_validate (const GstTensorInfo * info)
 
 /**
  * @brief Compare tensor info
- * @param TRUE if equal
+ * @return TRUE if equal, FALSE if given tensor infos are invalid or not equal.
  */
 gboolean
 gst_tensor_info_is_equal (const GstTensorInfo * i1, const GstTensorInfo * i2)
 {
   guint i;
 
-  g_return_val_if_fail (i1 != NULL, FALSE);
-  g_return_val_if_fail (i2 != NULL, FALSE);
+  if (gst_tensor_info_validate (i1) == FALSE ||
+      gst_tensor_info_validate (i2) == FALSE) {
+    return FALSE;
+  }
 
   if (i1->type != i2->type) {
     return FALSE;
@@ -301,7 +303,7 @@ gst_tensors_info_validate (const GstTensorsInfo * info)
 
 /**
  * @brief Compare tensors info
- * @param TRUE if equal
+ * @return TRUE if equal, FALSE if given tensor infos are invalid or not equal.
  */
 gboolean
 gst_tensors_info_is_equal (const GstTensorsInfo * i1, const GstTensorsInfo * i2)
@@ -311,7 +313,7 @@ gst_tensors_info_is_equal (const GstTensorsInfo * i1, const GstTensorsInfo * i2)
   g_return_val_if_fail (i1 != NULL, FALSE);
   g_return_val_if_fail (i2 != NULL, FALSE);
 
-  if (i1->num_tensors != i2->num_tensors) {
+  if (i1->num_tensors != i2->num_tensors || i1->num_tensors < 1) {
     return FALSE;
   }
 
@@ -952,7 +954,7 @@ gst_tensor_get_dimension_string (const tensor_dim dim)
 gulong
 gst_tensor_get_element_count (const tensor_dim dim)
 {
-  gsize count = 1;
+  gulong count = 1;
   guint i;
 
   for (i = 0; i < NNS_TENSOR_RANK_LIMIT; i++) {