diff --git a/config.pbtxt b/config.pbtxt
index b88d17f..874dc91 100644
--- a/config.pbtxt
+++ b/config.pbtxt
@@ -9,6 +9,22 @@ input [
     name: "Conv1_input"
     data_type: TYPE_FP32
     dims: [ -1, 28, 28, 1 ]
+  },
+  {
+    name: "Conv2_input"
+    data_type: TYPE_UINT16
+    dims: [ -1, 16 ]
+    reshape {
+      shape: [ 1, 16 ]
+    }
+  },
+  {
+    name: "Conv3_input"
+    data_type: TYPE_UINT16
+    dims: [ 123 ]
+    reshape {
+      shape: [ 123 ]
+    }
   }
 ]
 
@@ -17,6 +33,14 @@ output [
     name: "Softmax"
     data_type: TYPE_FP32
     dims: [ -1, 10 ]
+  },
+  {
+    name: "Softmax2"
+    data_type: TYPE_UINT16
+    dims: [ 1 ]
+    reshape {
+      shape: [ 1 ]
+    }
   }
 ]