Home
last modified time | relevance | path

Searched defs:SplitDynamicNodesHelper (Results 1 – 1 of 1) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/vm/
Dgraph_partition.cc514 struct SplitDynamicNodesHelper { struct
515 void AddNode(const AnfNodePtr &node, bool is_dynamic_shape) { in AddNode()
526 …s(std::vector<GraphSegmentPtr> *segments, std::map<AnfNodePtr, GraphSegmentPtr> *node_to_segment) { in AddSegments()
545 std::vector<AnfNodePtr> pre_nodes;
546 std::vector<AnfNodePtr> pre_dynamic_nodes;
547 std::vector<AnfNodePtr> pre_common_nodes;
548 std::set<AnfNodePtr> pre_common_nodes_set;
549 std::set<AnfNodePtr> pre_dynamic_nodes_set;
550 size_t merge_node_threshold = 6;