File tree 1 file changed +6
-6
lines changed
paddle/fluid/inference/api
1 file changed +6
-6
lines changed Original file line number Diff line number Diff line change @@ -58,18 +58,18 @@ class PaddleInferenceAnakinPredictor : public PaddlePredictor {
58
58
59
59
~PaddleInferenceAnakinPredictor () override ;
60
60
61
- static std::mutex mutex_;
62
- AnakinConfig config_;
63
- std::shared_ptr<anakin::Context<T>> ctx_p_;
64
- std::shared_ptr<anakin::graph::Graph<T, P>> graph_p_;
65
- anakin::Net<T, P, R>* executor_p_{nullptr };
66
-
61
+ protected:
67
62
void InitEnv ();
68
63
void InitGraph ();
69
64
virtual void OptimizeGraph ();
70
65
virtual void InitNet ();
71
66
virtual void SetContext ();
72
67
virtual void Predict ();
68
+ static std::mutex mutex_;
69
+ AnakinConfig config_;
70
+ std::shared_ptr<anakin::Context<T>> ctx_p_;
71
+ std::shared_ptr<anakin::graph::Graph<T, P>> graph_p_;
72
+ anakin::Net<T, P, R>* executor_p_{nullptr };
73
73
74
74
private:
75
75
bool RunImpl (const std::vector<PaddleTensor>& inputs,
You can’t perform that action at this time.
0 commit comments