diff --git a/config.pbtxt b/config.pbtxt index 816ff69..ef50f08 100644 --- a/config.pbtxt +++ b/config.pbtxt @@ -3,9 +3,35 @@ backend: "python" max_batch_size: 0 input [ + { + name: "tensor" + data_type: TYPE_BOOL + dims: [ 1 ] + }, + { + name: "tensor2" + data_type: TYPE_INT32 + dims: [ 1,2 ] + reshape { + shape: [ 1,1 ] + } + } ] output [ + { + name: "output1" + data_type: TYPE_UINT8 + dims: [ 1, 23 ] + }, + { + name: "output2" + data_type: TYPE_UINT32 + dims: [ 1, 30 ] + reshape { + shape: [ 1, 30 ] + } + } ]