Home
last modified time | relevance | path

Searched refs:GraphSegment (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/mindspore/core/ir/
Danf.h666 struct GraphSegment { struct
667GraphSegment(const std::vector<AnfNodePtr> &nodes, bool is_cut) : nodes_(nodes), is_cut_(is_cut) {} in GraphSegment() function
668 …void AddPreSegment(const std::shared_ptr<GraphSegment> &segment) { (void)pre_segments_.insert(segm… in AddPreSegment() argument
670 std::set<std::shared_ptr<GraphSegment>> pre_segments_; argument
674 using GraphSegmentPtr = std::shared_ptr<GraphSegment>;
/third_party/mindspore/mindspore/ccsrc/vm/
Dgraph_partition.cc507 auto segment = std::make_shared<GraphSegment>(nodes, false); in AddSegment()
701 auto segment = std::make_shared<GraphSegment>(segment_nodes, true); in Partition()
/third_party/mindspore/mindspore/ccsrc/pipeline/jit/
Dpipeline.cc1243 auto segment = std::make_shared<GraphSegment>(std::vector<AnfNodePtr>{app_init}, false); in InitExecDatasetVm()