|
|
|
@@ -151,7 +151,7 @@ class SessionBasic : public std::enable_shared_from_this<SessionBasic> { |
|
|
|
std::vector<AnfNodePtr> CreateValueNode(const CNodePtr &cnode, KernelGraph *graph); |
|
|
|
void CreateCNodeInputs(const CNodePtr &cnode, KernelGraph *graph, std::vector<AnfNodePtr> *cnode_inputs); |
|
|
|
std::vector<AnfNodePtr> CreateCallSwitchInputs(const CNodePtr &cnode, KernelGraph *graph); |
|
|
|
void GetCNodeInfo(const CNodePtr &cnode, std::vector<AnfNodePtr> *cnode_inputs); |
|
|
|
void GetCNodeInfo(const CNodePtr &cnode, std::vector<AnfNodePtr> *cnode_inputs) const; |
|
|
|
void GetNewCNodeInputs(const CNodePtr &cnode, KernelGraph *graph, std::vector<AnfNodePtr> *cnode_inputs, |
|
|
|
std::unordered_map<AnfNodePtr, AnfNodePtr> *other_graph_cnode); |
|
|
|
std::vector<AnfNodePtr> CreateCallSwitchLayerInputs(const CNodePtr &cnode, KernelGraph *graph); |
|
|
|
|