diff --git a/config.pbtxt b/config.pbtxt index 6c85282..e2f1536 100644 --- a/config.pbtxt +++ b/config.pbtxt @@ -4,9 +4,25 @@ platform: "tensorflow_savedmodel" max_batch_size: 0 input [ + { + name: "test1" + data_type: TYPE_UINT8 + dims: [ 123 ] + reshape { + shape: [ 123 ] + } + } ] output [ + { + name: "test2" + data_type: TYPE_UINT16 + dims: [ 321 ] + reshape { + shape: [ 321 ] + } + } ]