diff options
Diffstat (limited to 'caffe2/opt/onnxifi_transformer.h')
| -rw-r--r-- | caffe2/opt/onnxifi_transformer.h | 3 |
1 files changed, 2 insertions, 1 deletions
diff --git a/caffe2/opt/onnxifi_transformer.h b/caffe2/opt/onnxifi_transformer.h index b178909b22..a7ba90a0f9 100644 --- a/caffe2/opt/onnxifi_transformer.h +++ b/caffe2/opt/onnxifi_transformer.h @@ -79,7 +79,8 @@ class CAFFE2_API OnnxifiTransformer final { const std::string& onnx_model_str, const std::unordered_map<std::string, TensorShape>& output_size_hints, const std::unordered_set<std::string>& initialization_list, - const caffe2::NetDef& net); + const std::vector<std::string>& external_inputs, + const std::vector<std::string>& external_outputs); CaffeMap<std::string, TensorShape> SsaRewriteAndMapNames( Workspace* ws, |