Home
last modified time | relevance | path

Searched defs:subgraph (Results 1 – 25 of 31) sorted by relevance

12

/third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/graph/
Dsubgraph_node_pass.cc31 STATUS SubgraphNodePass::GetSubgraphAllTensorIndices(const std::unique_ptr<SubGraphT> &subgraph, in GetSubgraphAllTensorIndices()
54 const std::unique_ptr<SubGraphT> &subgraph) { in IsNodeInputInSubgraph()
62 const std::unique_ptr<SubGraphT> &subgraph) { in IsNodeOutputInSubgraph()
69 for (auto &subgraph : graph->subGraph) { in DecreaseSubgraphNodeIndices() local
81 for (auto &subgraph : graph->subGraph) { in IncreaseSubgraphNodeIndices() local
101 for (auto &subgraph : graph->subGraph) { in Run() local
121 for (auto &subgraph : graph->subGraph) { in Run() local
135 for (auto subgraph : contain_node_input_subgraphs) { in Run() local
Dsubgraph_tensor_pass.cc39 for (const auto &subgraph : graph->subGraph) { in IsUsing() local
51 for (const auto &subgraph : graph->subGraph) { in UpdateTensorIdx() local
Dtopological_sort_pass.cc33 for (auto &subgraph : graph->subGraph) { in Run() local
Dinfershape_pass.cc316 auto &subgraph = graph->subGraph.at(subgraph_index); in CopyPartialShapeToSubGraph() local
347 auto &subgraph = graph->subGraph.at(subgraph_index); in RestoreSubGraphInput() local
565 auto &subgraph = graph->subGraph.at(subgraph_index); in InitSearchTensor() local
/third_party/mindspore/mindspore/lite/src/delegate/npu/pass/
Dnpu_pass_manager.cc27 int NPUPassManager::RunPass(NPUGraph *subgraph) { in RunPass()
Dnpu_transform_pass.cc160 int NPUTransformPass::Run(NPUGraph *subgraph) { in Run()
Dnpu_insert_transform_pass.cc252 int NPUInsertTransformPass::Run(NPUGraph *subgraph) { in Run()
Dnpu_fusion_pass.cc444 int NPUFusionPass::Run(NPUGraph *subgraph) { in Run()
/third_party/mindspore/mindspore/_extends/graph_kernel/model/
Dmodel_builder.py172 def add_stitch_info(self, subgraph, desc): argument
181 def add_recompute_ops(self, subgraph, desc): argument
200 def dump(self, subgraph): argument
/third_party/mindspore/mindspore/lite/src/runtime/
Druntime_pass.cc138 bool RuntimePassValid(kernel::SubGraphKernel *subgraph) { in RuntimePassValid()
168 kernel::SubGraphKernel *subgraph = reinterpret_cast<kernel::SubGraphKernel *>(kernel); in Nc4hw4PassAct() local
/third_party/mindspore/mindspore/lite/src/
Dsub_graph_split.cc378 void SearchSubGraph::InsertNode(uint32_t index, Subgraph *subgraph, uint32_t last_index) { in InsertNode()
477 void SearchSubGraph::InsertHeadNode(uint32_t head_node_index, Subgraph *subgraph) { in InsertHeadNode()
500 void SearchSubGraph::InsertNodeByMid(uint32_t node_index, Subgraph *subgraph, uint32_t last_index) { in InsertNodeByMid()
635 Subgraph subgraph; in InitSearchSubGraphByOutput() local
881 for (auto &subgraph : subgraphs) { in SubGraphSplitByOffLineParallel() local
924 void SearchSubGraph::InsertParallelNode(uint32_t index, Subgraph *subgraph) { in InsertParallelNode()
Dscheduler.cc629 auto subgraph = src_model_->graph_.sub_graphs_.at(subgraph_index); in RestoreSubGraphInput() local
646 auto subgraph = src_model_->graph_.sub_graphs_.at(subgraph_index); in CopyPartialShapeToSubGraph() local
730 auto subgraph = src_model_->graph_.sub_graphs_.at(subgraph_index); in InferSubGraphShape() local
1264 auto subgraph = src_model_->graph_.sub_graphs_.at(subgraph_index); in SubGraphPreferDataType() local
1418 auto subgraph = src_model_->graph_.sub_graphs_.at(subgraph_index); in ScheduleSubGraphToKernels() local
1564 auto subgraph = FindAllSubGraphKernels(src_kernel, *context_, &index, schema_version_); in ConstructSubGraphs() local
Dlite_mindrt.cc241 …auto subgraph = reinterpret_cast<kernel::PartialFusionKernel *>(partial_node->kernel())->subgraph_… in CompileArrowThroughPartialCall() local
452 …auto subgraph = static_cast<kernel::PartialFusionKernel *>(true_partial_node_->kernel())->subgraph… in CompileTrueBranchArrow() local
477 …auto subgraph = static_cast<kernel::PartialFusionKernel *>(false_partial_node_->kernel())->subgrap… in CompileFalseBranchArrow() local
Dlite_model.h165 auto *subgraph = new (std::nothrow) LiteGraph::SubGraph(); in MetaGraphMappingSubGraph() local
Dlite_session.cc440 for (auto subgraph : kernels_) { in IsolateOutputTensor() local
485 auto subgraph = reinterpret_cast<kernel::SubGraphKernel *>(kernel); in FreePackOpWeight() local
Dlite_model.cc154 auto *subgraph = new (std::nothrow) LiteGraph::SubGraph(); in ConvertSubGraph() local
/third_party/mindspore/mindspore/lite/micro/coder/
Dtrain.cc30 std::set<OperatorCoder *> subgraph; in FindInferenceOpcoders() local
/third_party/mindspore/mindspore/lite/src/delegate/nnrt/
Dnnrt_delegate.cc320 for (auto subgraph : lite_graph.sub_graphs_) { in ShallowCopyLiteGraph() local
/third_party/boost/boost/graph/
Dsubgraph.hpp82 template < typename Graph > class subgraph class
119 subgraph() : m_parent(0), m_edge_counter(0) {} in subgraph() function in boost::subgraph
121 subgraph(const graph_property_type& p) in subgraph() function in boost::subgraph
126 subgraph(vertices_size_type n, in subgraph() function in boost::subgraph
137 subgraph(const subgraph& x) : m_parent(x.m_parent), m_edge_counter(0) in subgraph() function in boost::subgraph
/third_party/mindspore/mindspore/lite/src/delegate/npu/
Dnpu_graph.cc256 auto subgraph = new (std::nothrow) NPUSubGraph(npu_ops, npu_manager_); in CreateNPUSubgraphKernel() local
/third_party/mindspore/mindspore/lite/tools/converter/parser/onnx/
Donnx_model_parser.cc568 for (auto &subgraph : all_subgraphs_) { in Parse() local
733 …odelParser::ConvertIfSubgraph(const onnx::GraphProto &subgraph_proto, const FuncGraphPtr &subgraph, in ConvertIfSubgraph()
792 FuncGraphPtr subgraph = nullptr; in ConvertIfOnnxNode() local
/third_party/mindspore/mindspore/lite/tools/common/
Dgraph_util.cc322 for (auto &subgraph : graphT->subGraph) { in RemoveTensor() local
676 for (auto &subgraph : meta_graphT->subGraph) { in SetSubgraphTensorIndices() local
Dfunc_graph_subgraph.cc296 bool SubGraph::MergeSubGraph(const SubGraphPtr &subgraph) { in MergeSubGraph()
/third_party/mindspore/mindspore/lite/src/train/
Dtrain_export.cc449 for (auto &subgraph : meta_graph_->subGraph) { in TopologicalSort() local
/third_party/boost/libs/graph_parallel/test/
Dssca.cpp379 std::vector<vertex_descriptor>& subgraph = subgraphs.back(); in subgraph_extraction() local
474 std::vector<vertex_descriptor>& subgraph = subgraphs.back(); in seq_subgraph_extraction() local

12