/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/predictor
NameSizeModeActions
InferenceGraph.h8330644editdlrm
predictor.h18100644editdlrm
predictor_config.h19340644editdlrm
predictor_utils.h8020644editdlrm
ThreadLocalPtr.h41110644editdlrm
transforms.h2410644editdlrm
Edit: /usr/local/lib64/python3.6/site-packages/torch/include/caffe2/predictor/predictor_utils.h (802B)
#pragma once #include "caffe2/core/db.h" #include "caffe2/core/workspace.h" #include "caffe2/predictor/predictor_config.h" #include "caffe2/proto/metanet.pb.h" namespace caffe2 { namespace predictor_utils { TORCH_API const NetDef& getNet(const MetaNetDef& def, const std::string& name); const ::google::protobuf::RepeatedPtrField<::std::string>& getBlobs( const MetaNetDef& def, const std::string& name); TORCH_API std::unique_ptr extractMetaNetDef( db::Cursor* cursor, const std::string& key); // Extract the MetaNetDef from `db`, and run the global init net on the // `master` workspace. TORCH_API std::unique_ptr runGlobalInitialization( std::unique_ptr db, Workspace* master); } // namespace predictor_utils } // namespace caffe2