diff options
Diffstat (limited to 'caffe2/python/pybind_state.cc')
| -rw-r--r-- | caffe2/python/pybind_state.cc | 4 |
1 files changed, 0 insertions, 4 deletions
diff --git a/caffe2/python/pybind_state.cc b/caffe2/python/pybind_state.cc index d4f54d8a87..609ee01e69 100644 --- a/caffe2/python/pybind_state.cc +++ b/caffe2/python/pybind_state.cc @@ -1604,9 +1604,7 @@ void addGlobalMethods(py::module& m) { m.def( "onnxifi", [](const py::bytes& pred_net_str, - const std::vector<std::string>& external_inputs, const std::unordered_map<std::string, std::vector<int>>& shapes, - bool infer_shapes, int max_batch_size, int max_seq_size, bool debug_builder, @@ -1622,7 +1620,6 @@ void addGlobalMethods(py::module& m) { it.first, CreateTensorShape(it.second, TensorProto::FLOAT)); } OnnxifiTransformerOptions opts; - opts.infer_shapes = infer_shapes; opts.bound_shape_spec.max_batch_size = max_batch_size; opts.bound_shape_spec.max_seq_size = max_seq_size; opts.debug = debug_builder; @@ -1633,7 +1630,6 @@ void addGlobalMethods(py::module& m) { ts.Transform( curr_ws, &pred_net, - external_inputs, weight_names, tensor_shapes, std::unordered_set<int>()); |