diff --git a/config.pbtxt b/config.pbtxt
index febb81f..adc1d83 100644
--- a/config.pbtxt
+++ b/config.pbtxt
@@ -5,9 +5,49 @@ platform: "tensorflow_savedmodel"
 max_batch_size: 0
 
 input [
+  {
+    name: "test1"
+    data_type: TYPE_UINT8
+    dims: [ 1 ]
+    reshape {
+      shape: [ 1 ]
+    }
+  },
+  {
+    name: "test3"
+    data_type: TYPE_UINT16
+    dims: [ 3 ]
+    reshape {
+      shape: [ 3 ]
+    }
+  },
+  {
+    name: "test5"
+    data_type: TYPE_UINT8
+    dims: [ 5 ]
+    reshape {
+      shape: [ 5 ]
+    }
+  }
 ]
 
 output [
+  {
+    name: "test2"
+    data_type: TYPE_UINT16
+    dims: [ 2 ]
+    reshape {
+      shape: [ 2 ]
+    }
+  },
+  {
+    name: "test4"
+    data_type: TYPE_UINT8
+    dims: [ 4 ]
+    reshape {
+      shape: [ 4 ]
+    }
+  }
 ]