Searched defs:subgraph_index (Results 1 – 9 of 9) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/legacy_optimizer/graph/ |
| D | infershape_pass.cc | 386 auto subgraph_index = PartialGraphIndex(partial_node); in CopyPartialShapeToSubGraph() local
419 auto subgraph_index = PartialGraphIndex(partial_node); in RestoreSubGraphInput() local
431 auto subgraph_index = PartialGraphIndex(partial_node); in SetNonTailCallOutputShape() local
452 int64_t subgraph_index = PartialGraphIndex(partial_node); in InferPartialNode() local
578 int InferShapePass::InferSubgraph(const int64_t &subgraph_index, MetaGraphT *graph) { in InferSubgraph()
654 int InferShapePass::InitSearchTensor(const int64_t &subgraph_index, MetaGraphT *graph, in InitSearchTensor()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/litert/ |
| D | scheduler.cc | 772 …auto subgraph_index = GetPartialGraphIndex(partial_node->primitive_, context_->get_schema_version(… in RestoreSubGraphInput() local
790 …auto subgraph_index = GetPartialGraphIndex(partial_node->primitive_, context_->get_schema_version(… in CopyPartialShapeToSubGraph() local
820 int subgraph_index = GetPartialGraphIndex(node->primitive_, context_->get_schema_version()); in InferPartialShape() local
866 int Scheduler::InferSubGraphShape(size_t subgraph_index) { in InferSubGraphShape()
1341 auto subgraph_index = GetPartialGraphIndex(src_node->primitive_, context_->get_schema_version()); in SchedulePartialToKernel() local
1358 int Scheduler::SubGraphPreferDataType(const int &subgraph_index, TypeId *prefer_data_type) { in SubGraphPreferDataType()
1416 kernel::KernelExec *Scheduler::SchedulePartialToSubGraphKernel(const int &subgraph_index) { in SchedulePartialToSubGraphKernel()
1447 …tor<kernel::KernelExec *> Scheduler::ScheduleSubGraphToSubGraphKernels(const int &subgraph_index) { in ScheduleSubGraphToSubGraphKernels()
1535 int Scheduler::ScheduleSubGraphToKernels(size_t subgraph_index, std::vector<kernel::KernelExec *> *… in ScheduleSubGraphToKernels()
|
| D | runtime_shape_fusion_pass.cc | 105 int ShapeFusionPass::FusePostNodes(LiteGraph::Node *node, size_t subgraph_index) { in FusePostNodes()
181 uint32_t input_idx, size_t subgraph_index) { in CheckCanFused()
227 std::vector<uint32_t> *input_indices, size_t subgraph_index) { in DoFuse()
|
| D | runtime_shape_fusion_pass.h | 101 void Run(LiteGraph::Node *node, size_t subgraph_index) { in Run()
|
| D | lite_model.cc | 184 int64_t subgraph_index = partial_fusion->sub_graph_index(); in NodeVerify() local
|
| D | sub_graph_split.cc | 203 const schema::Primitive *SearchSubGraph::CreatePartialPrimitive(int64_t subgraph_index) { in CreatePartialPrimitive()
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/converter/parser/tf/ |
| D | functionalize_control_op_pass.h | 60 static int subgraph_index = 1; in GetSubgraphIndex() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/tools/lite_exporter/ |
| D | anf_exporter.cc | 361 const size_t &subgraph_index) { in SetSubGraphInputIndex()
389 int AnfExporter::SetSubGraphOutputIndex(const CNodePtr &cnode, const size_t subgraph_index, in SetSubGraphOutputIndex()
614 … const size_t &subgraph_index, const bool &keep_graph, const bool ©_primitive) { in Anf2Fb()
735 auto subgraph_index = meta_graphT->subGraph.size() - 1; in ExportSubgraph() local
|
| /third_party/mindspore/mindspore-src/source/mindspore/lite/src/control_flow/ |
| D | control_flow_scheduler.cc | 296 void ControlFlowScheduler::RecordSubgraphCaller(const size_t &subgraph_index, kernel::KernelExec *p… in RecordSubgraphCaller()
751 auto &subgraph_index = pair.second; in SetSubgraphForPartialNode() local
|