-
Notifications
You must be signed in to change notification settings - Fork 655
Expand file tree
/
Copy pathpredict.proto
More file actions
29 lines (22 loc) · 803 Bytes
/
predict.proto
File metadata and controls
29 lines (22 loc) · 803 Bytes
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
// SPDX-License-Identifier: Apache-2.0
syntax = "proto3";
import "onnx-ml.proto";
package onnxruntime.server;
// PredictRequest specifies how inputs are mapped to tensors
// and how outputs are filtered before returning to user.
message PredictRequest {
reserved 1;
// Input Tensors.
// This is a mapping between output name and tensor.
map<string, onnx.TensorProto> inputs = 2;
// Output Filters.
// This field is to specify which output fields need to be returned.
// If the list is empty, all outputs will be included.
repeated string output_filter = 3;
}
// Response for PredictRequest on successful run.
message PredictResponse {
// Output Tensors.
// This is a mapping between output name and tensor.
map<string, onnx.TensorProto> outputs = 1;
}