File tree Expand file tree Collapse file tree 2 files changed +8
-8
lines changed
ensemble_models/mix_platform_float32_float32_float32 Expand file tree Collapse file tree 2 files changed +8
-8
lines changed Original file line number Diff line number Diff line change @@ -238,7 +238,7 @@ function generate_model_repository() {
238
238
239
239
ENSEMBLE_MODELS=" ${ENSEMBLE_MODELS} batch_to_nobatch_float32_float32_float32 batch_to_nobatch_nobatch_float32_float32_float32 nobatch_to_batch_float32_float32_float32 nobatch_to_batch_nobatch_float32_float32_float32 mix_nobatch_batch_float32_float32_float32"
240
240
241
- if [[ $BACKENDS == * " plan " * ]] ; then
241
+ if [[ $BACKENDS == * " libtorch " * ]] ; then
242
242
ENSEMBLE_MODELS=" ${ENSEMBLE_MODELS} mix_platform_float32_float32_float32 mix_ensemble_int32_float32_float32"
243
243
fi
244
244
Original file line number Diff line number Diff line change @@ -70,7 +70,7 @@ ensemble_scheduling {
70
70
}
71
71
},
72
72
{
73
- model_name: " plan_float32_float32_float32 "
73
+ model_name: " libtorch_float32_float32_float32 "
74
74
model_version: 1
75
75
input_map {
76
76
key: " INPUT0"
@@ -81,12 +81,12 @@ ensemble_scheduling {
81
81
value: " INPUT1"
82
82
}
83
83
output_map {
84
- key: " OUTPUT0 "
84
+ key: " OUTPUT__0 "
85
85
value: " double_input1"
86
86
}
87
87
},
88
88
{
89
- model_name: " plan_float32_float32_float32 "
89
+ model_name: " libtorch_float32_float32_float32 "
90
90
model_version: 1
91
91
input_map {
92
92
key: " INPUT0"
@@ -97,7 +97,7 @@ ensemble_scheduling {
97
97
value: " INPUT0"
98
98
}
99
99
output_map {
100
- key: " OUTPUT1 "
100
+ key: " OUTPUT__1 "
101
101
value: " input0_val"
102
102
}
103
103
},
@@ -118,7 +118,7 @@ ensemble_scheduling {
118
118
}
119
119
},
120
120
{
121
- model_name: " plan_float32_float32_float32 "
121
+ model_name: " libtorch_float32_float32_float32 "
122
122
model_version: 1
123
123
input_map {
124
124
key: " INPUT0"
@@ -129,11 +129,11 @@ ensemble_scheduling {
129
129
value: " input1_val"
130
130
}
131
131
output_map {
132
- key: " OUTPUT0 "
132
+ key: " OUTPUT__0 "
133
133
value: " OUTPUT0"
134
134
}
135
135
output_map {
136
- key: " OUTPUT1 "
136
+ key: " OUTPUT__1 "
137
137
value: " OUTPUT1"
138
138
}
139
139
}
You can’t perform that action at this time.
0 commit comments