apps/FastDeploy
1
0
Fork 0
mirror of https://github.com/PaddlePaddle/FastDeploy.git synced 2025-10-23 16:44:22 +08:00
Code Issues Actions 5 Packages Projects Releases Wiki Activity

Add time statis functions for fastdeploy model (#96)

Browse Source 
* Add time statis functions for fastdeploy model

* Create README.md

* Update README.md

* Update fastdeploy_model.cc

* Update fastdeploy_model.cc
This commit is contained in:
Jason
2022-08-11 14:06:43 +08:00
committed by GitHub
parent 7ccb0c9ff5
commit a7cf09354e
8 changed files with 112 additions and 20 deletions
Download Patch File Download Diff File Expand all files Collapse all files

9
csrc/fastdeploy/pybind/fastdeploy_model.cc
View File

@@ -12,8 +12,8 @@
// See the License for the specific language governing permissions and
// limitations under the License.
#include "fastdeploy/pybind/main.h"
#include "fastdeploy/fastdeploy_model.h"
#include "fastdeploy/pybind/main.h"
namespace fastdeploy {
@@ -25,6 +25,11 @@ void BindFDModel(pybind11::module& m) {
.def("num_outputs_of_runtime", &FastDeployModel::NumOutputsOfRuntime)
.def("input_info_of_runtime", &FastDeployModel::InputInfoOfRuntime)
.def("output_info_of_runtime", &FastDeployModel::OutputInfoOfRuntime)
.def("enable_record_time_of_runtime", &FastDeployModel::EnableRecordTimeOfRuntime)
.def("disable_record_time_of_runtime",
&FastDeployModel::DisableRecordTimeOfRuntime)
.def("print_statis_info_of_runtime",
&FastDeployModel::PrintStatisInfoOfRuntime)
.def("initialized", &FastDeployModel::Initialized)
.def_readwrite("runtime_option", &FastDeployModel::runtime_option)
.def_readwrite("valid_cpu_backends", &FastDeployModel::valid_cpu_backends)
@@ -32,4 +37,4 @@ void BindFDModel(pybind11::module& m) {
&FastDeployModel::valid_gpu_backends);
}
} // namespace fastdeploy
} // namespace fastdeploy