未验证 提交 04ea7cb0 编写于 作者: 石晓伟 提交者: GitHub

modify the access level of anakin engine (#18015)

test=develop
上级 0bf25351
......@@ -58,18 +58,18 @@ class PaddleInferenceAnakinPredictor : public PaddlePredictor {
~PaddleInferenceAnakinPredictor() override;
static std::mutex mutex_;
AnakinConfig config_;
std::shared_ptr<anakin::Context<T>> ctx_p_;
std::shared_ptr<anakin::graph::Graph<T, P>> graph_p_;
anakin::Net<T, P, R>* executor_p_{nullptr};
protected:
void InitEnv();
void InitGraph();
virtual void OptimizeGraph();
virtual void InitNet();
virtual void SetContext();
virtual void Predict();
static std::mutex mutex_;
AnakinConfig config_;
std::shared_ptr<anakin::Context<T>> ctx_p_;
std::shared_ptr<anakin::graph::Graph<T, P>> graph_p_;
anakin::Net<T, P, R>* executor_p_{nullptr};
private:
bool RunImpl(const std::vector<PaddleTensor>& inputs,
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册