|
|
|
@ -17,6 +17,7 @@
|
|
|
|
|
#include <string>
|
|
|
|
|
#include "paddle/fluid/framework/scope.h"
|
|
|
|
|
#include "paddle/fluid/inference/analysis/analysis_pass.h"
|
|
|
|
|
#include "paddle/fluid/platform/place.h"
|
|
|
|
|
|
|
|
|
|
namespace paddle {
|
|
|
|
|
namespace inference {
|
|
|
|
@ -34,11 +35,10 @@ class IrGraphBuildPass : public AnalysisPass {
|
|
|
|
|
private:
|
|
|
|
|
std::unique_ptr<framework::ProgramDesc> LoadModel(
|
|
|
|
|
const std::string &path, framework::Scope *scope,
|
|
|
|
|
const boost::variant<CUDAPlace, CPUPlace, CUDAPinnedPlace> &place);
|
|
|
|
|
const platform::Place &place);
|
|
|
|
|
std::unique_ptr<framework::ProgramDesc> LoadModel(
|
|
|
|
|
const std::string &program_path, const std::string ¶ms_path,
|
|
|
|
|
framework::Scope *scope,
|
|
|
|
|
const boost::variant<CUDAPlace, CPUPlace, CUDAPinnedPlace> &place);
|
|
|
|
|
framework::Scope *scope, const platform::Place &place);
|
|
|
|
|
|
|
|
|
|
std::string model_binary_str_;
|
|
|
|
|
};
|
|
|
|
|