diff --git a/config.pbtxt b/config.pbtxt index 913efaf..9c9312e 100644 --- a/config.pbtxt +++ b/config.pbtxt @@ -1,13 +1,28 @@ # Triton backend to use -name: "fashion-mnist-2" backend: "tensorflow" -platform: "tensorflow_graphdef" +platform: "tensorflow_savedmodel" max_batch_size: 0 input [ + { + name: "test" + data_type: TYPE_UINT8 + dims: [ 123 ] + reshape { + shape: [ 123 ] + } + } ] output [ + { + name: "test2" + data_type: TYPE_UINT16 + dims: [ 321 ] + reshape { + shape: [ 321 ] + } + } ]