[Mux] accept flex tensor
authorJaeyun <jy1210.jung@samsung.com>
Mon, 26 Apr 2021 10:41:46 +0000 (19:41 +0900)
committerMyungJoo Ham <myungjoo.ham@samsung.com>
Fri, 28 May 2021 04:40:52 +0000 (13:40 +0900)
add pad template to support flex tensor in tensor-mux.

Signed-off-by: Jaeyun <jy1210.jung@samsung.com>
gst/nnstreamer/tensor_mux/gsttensormux.c

index 47b23d1..8193a14 100644 (file)
@@ -94,7 +94,12 @@ enum
 /**
  * @brief Default caps string for sink pad.
  */
-#define CAPS_STRING_SINK GST_TENSOR_CAP_DEFAULT "; " GST_TENSORS_CAP_DEFAULT
+#define CAPS_STRING_SINK GST_TENSOR_CAP_DEFAULT ";" GST_TENSORS_CAP_DEFAULT ";" GST_TENSORS_FLEX_CAP_DEFAULT
+
+/**
+ * @brief Default caps string for src pad.
+ */
+#define CAPS_STRING_SRC GST_TENSORS_CAP_DEFAULT ";" GST_TENSORS_FLEX_CAP_DEFAULT
 
 /**
  * @brief the capabilities of the inputs and outputs.
@@ -103,7 +108,7 @@ enum
 static GstStaticPadTemplate src_templ = GST_STATIC_PAD_TEMPLATE ("src",
     GST_PAD_SRC,
     GST_PAD_ALWAYS,
-    GST_STATIC_CAPS (GST_TENSORS_CAP_DEFAULT)
+    GST_STATIC_CAPS (CAPS_STRING_SRC)
     );
 
 static GstStaticPadTemplate sink_templ = GST_STATIC_PAD_TEMPLATE ("sink_%u",