Skip to content

Commit

Permalink
Add reshape tests for autocomplete (triton-inference-server#4121)
Browse files Browse the repository at this point in the history
* add reshape tests for autocomplete

* copy model files from QA model repo

* fix copyright
  • Loading branch information
krishung5 committed Mar 29, 2022
1 parent 380b88a commit e9f2187
Show file tree
Hide file tree
Showing 10 changed files with 512 additions and 1 deletion.
Original file line number Diff line number Diff line change
@@ -0,0 +1,41 @@
name: "reshape_config_provided"
version_policy {
latest {
num_versions: 1
}
}
max_batch_size: 8
input {
name: "INPUT0"
data_type: TYPE_FP32
dims: 1
reshape {
}
}
input {
name: "INPUT1"
data_type: TYPE_FP32
dims: 8
reshape {
shape: 4
shape: 1
shape: 2
}
}
output {
name: "OUTPUT0"
data_type: TYPE_FP32
dims: 1
reshape {
}
}
output {
name: "OUTPUT1"
data_type: TYPE_FP32
dims: 8
reshape {
shape: 4
shape: 1
shape: 2
}
}
Original file line number Diff line number Diff line change
@@ -0,0 +1,58 @@
name: "reshape_config_provided"
platform: "tensorflow_graphdef"
version_policy {
latest {
num_versions: 1
}
}
max_batch_size: 8
input {
name: "INPUT0"
data_type: TYPE_FP32
dims: 1
reshape {
}
}
input {
name: "INPUT1"
data_type: TYPE_FP32
dims: 8
reshape {
shape: 4
shape: 1
shape: 2
}
}
output {
name: "OUTPUT0"
data_type: TYPE_FP32
dims: 1
reshape {
}
}
output {
name: "OUTPUT1"
data_type: TYPE_FP32
dims: 8
reshape {
shape: 4
shape: 1
shape: 2
}
}
instance_group {
name: "reshape_config_provided"
count: 1
gpus: 0
kind: KIND_GPU
}
default_model_filename: "model.graphdef"
optimization {
input_pinned_memory {
enable: true
}
output_pinned_memory {
enable: true
}
}
backend: "tensorflow"
Original file line number Diff line number Diff line change
@@ -0,0 +1,34 @@
name: "reshape_config_provided"
max_batch_size: 8
input [
{
name: "INPUT0"
data_type: TYPE_FP32
dims: [ 1 ]
reshape: { shape: [ ] }
}
]
output [
{
name: "OUTPUT0"
data_type: TYPE_FP32
dims: [ 1 ]
reshape: { shape: [ ] }
}
]
input [
{
name: "INPUT1"
data_type: TYPE_FP32
dims: [ 8 ]
reshape: { shape: [ 4,1,2 ] }
}
]
output [
{
name: "OUTPUT1"
data_type: TYPE_FP32
dims: [ 8 ]
reshape: { shape: [ 4,1,2 ] }
}
]
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
name: "reshape_config_provided"
platform: "tensorflow_savedmodel"
version_policy {
latest {
num_versions: 1
}
}
max_batch_size: 8
input {
name: "INPUT1"
data_type: TYPE_FP32
dims: 4
dims: 1
dims: 2
}
input {
name: "INPUT0"
data_type: TYPE_FP32
dims: 1
reshape {
}
}
output {
name: "OUTPUT1"
data_type: TYPE_FP32
dims: 4
dims: 1
dims: 2
}
output {
name: "OUTPUT0"
data_type: TYPE_FP32
dims: 1
reshape {
}
}
instance_group {
name: "reshape_config_provided"
count: 1
gpus: 0
kind: KIND_GPU
}
default_model_filename: "model.savedmodel"
optimization {
input_pinned_memory {
enable: true
}
output_pinned_memory {
enable: true
}
}
backend: "tensorflow"
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
name: "reshape_config_provided"
platform: "tensorflow_savedmodel"
version_policy {
latest {
num_versions: 1
}
}
max_batch_size: 8
input {
name: "INPUT0"
data_type: TYPE_FP32
dims: 1
reshape {
}
}
input {
name: "INPUT1"
data_type: TYPE_FP32
dims: 4
dims: 1
dims: 2
}
output {
name: "OUTPUT1"
data_type: TYPE_FP32
dims: 4
dims: 1
dims: 2
}
output {
name: "OUTPUT0"
data_type: TYPE_FP32
dims: 1
reshape {
}
}
instance_group {
name: "reshape_config_provided"
count: 1
gpus: 0
kind: KIND_GPU
}
default_model_filename: "model.savedmodel"
optimization {
input_pinned_memory {
enable: true
}
output_pinned_memory {
enable: true
}
}
backend: "tensorflow"
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
name: "reshape_config_provided"
platform: "tensorflow_savedmodel"
version_policy {
latest {
num_versions: 1
}
}
max_batch_size: 8
input {
name: "INPUT1"
data_type: TYPE_FP32
dims: 4
dims: 1
dims: 2
}
input {
name: "INPUT0"
data_type: TYPE_FP32
dims: 1
reshape {
}
}
output {
name: "OUTPUT0"
data_type: TYPE_FP32
dims: 1
reshape {
}
}
output {
name: "OUTPUT1"
data_type: TYPE_FP32
dims: 4
dims: 1
dims: 2
}
instance_group {
name: "reshape_config_provided"
count: 1
gpus: 0
kind: KIND_GPU
}
default_model_filename: "model.savedmodel"
optimization {
input_pinned_memory {
enable: true
}
output_pinned_memory {
enable: true
}
}
backend: "tensorflow"
Original file line number Diff line number Diff line change
@@ -0,0 +1,52 @@
name: "reshape_config_provided"
platform: "tensorflow_savedmodel"
version_policy {
latest {
num_versions: 1
}
}
max_batch_size: 8
input {
name: "INPUT0"
data_type: TYPE_FP32
dims: 1
reshape {
}
}
input {
name: "INPUT1"
data_type: TYPE_FP32
dims: 4
dims: 1
dims: 2
}
output {
name: "OUTPUT0"
data_type: TYPE_FP32
dims: 1
reshape {
}
}
output {
name: "OUTPUT1"
data_type: TYPE_FP32
dims: 4
dims: 1
dims: 2
}
instance_group {
name: "reshape_config_provided"
count: 1
gpus: 0
kind: KIND_GPU
}
default_model_filename: "model.savedmodel"
optimization {
input_pinned_memory {
enable: true
}
output_pinned_memory {
enable: true
}
}
backend: "tensorflow"
Loading

0 comments on commit e9f2187

Please sign in to comment.