Skip to content
Snippets Groups Projects

Updatedoc

Merged Nathan Thoumine requested to merge nthm/aidge_export_tensorrt:updatedoc into master
All threads resolved!
Files
2
@@ -23,7 +23,7 @@ void init_Graph(py::module& m)
@@ -23,7 +23,7 @@ void init_Graph(py::module& m)
py::arg("device_id") = 0,
py::arg("device_id") = 0,
py::arg("nb_bits") = -32)
py::arg("nb_bits") = -32)
.def("device", &Graph::device, py::arg("id"))
.def("device", &Graph::device, py::arg("id"))
.def("load", &Graph::load, py::arg("filepath"))
.def("load", &Graph::load, py::arg("filepath"))
.def("save", &Graph::save, py::arg("filepath"))
.def("save", &Graph::save, py::arg("filepath"))
.def("calibrate", &Graph::calibrate, py::arg("calibration_folder_path") = "./calibration_folder/", py::arg("cache_file_path") = "./calibration_cache", py::arg("batch_size") = 1)
.def("calibrate", &Graph::calibrate, py::arg("calibration_folder_path") = "./calibration_folder/", py::arg("cache_file_path") = "./calibration_cache", py::arg("batch_size") = 1)
Loading