Searched refs:SessionBasic (Results 1 – 15 of 15) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/backend/session/ |
| D | session_basic.cc | 59 MS_REG_SESSION(kSessionBasic, SessionBasic);
501 GraphId SessionBasic::graph_sum_ = 0;
503 void SessionBasic::InitExecutor(const std::string &device_name, uint32_t device_id) { in InitExecutor()
509 GraphId SessionBasic::GetGraphIdByNode(const AnfNodePtr &front_anf) const { in GetGraphIdByNode()
523 KernelGraphPtr SessionBasic::GetGraph(mindspore::GraphId graph_id) const { in GetGraph()
532 void SessionBasic::ClearGraph() { in ClearGraph()
541 void SessionBasic::InitInternalOutputParameter(const AnfNodePtr &out_node, const AnfNodePtr ¶me… in InitInternalOutputParameter()
590 AnfNodePtr SessionBasic::CreateParameterFromTuple(const AnfNodePtr &node, KernelGraph *graph) { in CreateParameterFromTuple()
627 ParameterPtr SessionBasic::CreateNewParameterFromParameter(const AnfNodePtr &anf, KernelGraph *grap… in CreateNewParameterFromParameter()
662 AnfNodePtr SessionBasic::CreateNewParameterFromCNode(const AnfNodePtr &anf, KernelGraph *graph) { in CreateNewParameterFromCNode()
[all …]
|
| D | session_factory.h | 28 using SessionCreator = std::function<std::shared_ptr<SessionBasic>()>;
33 std::shared_ptr<SessionBasic> Create(const std::string &device_name);
|
| D | cpu_session.h | 28 class CPUSession : public SessionBasic {
35 void UnifyMindIR(const KernelGraphPtr &graph) override { SessionBasic::UnifyMindIR(graph); } in UnifyMindIR()
|
| D | gpu_session.h | 32 class GPUSession : public SessionBasic {
40 void UnifyMindIR(const KernelGraphPtr &graph) override { SessionBasic::UnifyMindIR(graph); } in UnifyMindIR()
|
| D | session_basic.h | 93 class SessionBasic : public std::enable_shared_from_this<SessionBasic> {
95 SessionBasic() : context_(nullptr), summary_callback_(nullptr), device_id_(0) { in SessionBasic() function
104 virtual ~SessionBasic() { summary_callback_ = nullptr; } in ~SessionBasic()
335 using SessionPtr = std::shared_ptr<session::SessionBasic>;
|
| D | session_factory.cc | 33 std::shared_ptr<SessionBasic> SessionFactory::Create(const std::string &device_name) { in Create()
|
| D | ascend_session.h | 39 class AscendSession : public SessionBasic {
|
| D | ascend_session.cc | 382 SessionBasic::UnifyMindIR(graph); in UnifyMindIR()
1625 SessionBasic::SetSummaryNodes(graph); in RecurseSetSummaryNodes()
1637 SessionBasic::SetSummaryNodes(child_graph.get()); in RecurseSetSummaryNodes()
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/model/acl/acl_vm/ |
| D | acl_multi_graph_session.h | 30 class MultiGraphAclSession : public session::SessionBasic {
|
| D | acl_multi_graph_session.cc | 50 auto kernel_graph = SessionBasic::ConstructKernelGraph(lst, outputs, false); in CompileGraphImpl()
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/graph/gpu/ |
| D | gpu_graph_impl.h | 48 std::shared_ptr<session::SessionBasic> session_impl_;
|
| /third_party/mindspore/mindspore/ccsrc/cxx_api/graph/ascend/ |
| D | ascend_graph_impl.h | 52 std::shared_ptr<session::SessionBasic> session_impl_;
|
| /third_party/mindspore/mindspore/ccsrc/ps/ |
| D | parameter_server.h | 153 std::shared_ptr<session::SessionBasic> sess_;
|
| /third_party/mindspore/mindspore/ccsrc/pipeline/pynative/ |
| D | pynative_execute_ge.cc | 47 static std::shared_ptr<session::SessionBasic> session = nullptr;
|
| D | pynative_execute.cc | 84 std::shared_ptr<session::SessionBasic> kSession = nullptr;
|