Searched defs:subgraph_index (Results 1 – 6 of 6) sorted by relevance
| /third_party/mindspore/mindspore/lite/src/ |
| D | lite_model.cc | 65 …for (size_t subgraph_index = 0; subgraph_index < this->graph_.sub_graphs_.size(); ++subgraph_index… in ConvertAttrToTensors() local
83 …for (size_t subgraph_index = 0; subgraph_index < this->graph_.sub_graphs_.size(); ++subgraph_index… in ConvertAttrToTensors() local
90 for (auto subgraph_index : subgraphs_with_node) { in ConvertAttrToTensors() local
220 auto subgraph_index = GetPartialGraphIndex(node->primitive_, schema_version_); in NodeVerify() local
|
| D | scheduler.cc | 628 auto subgraph_index = GetPartialGraphIndex(partial_node->primitive_, schema_version_); in RestoreSubGraphInput() local
645 auto subgraph_index = GetPartialGraphIndex(partial_node->primitive_, schema_version_); in CopyPartialShapeToSubGraph() local
679 int subgraph_index = GetPartialGraphIndex(node->primitive_, schema_version_); in InferPartialShape() local
726 int Scheduler::InferSubGraphShape(size_t subgraph_index) { in InferSubGraphShape()
1247 auto subgraph_index = GetPartialGraphIndex(src_node->primitive_, schema_version_); in SchedulePartialToKernel() local
1258 int Scheduler::SubGraphPreferDataType(const int &subgraph_index, TypeId *prefer_data_type) { in SubGraphPreferDataType()
1313 kernel::LiteKernel *Scheduler::SchedulePartialToSubGraphKernel(const int &subgraph_index) { in SchedulePartialToSubGraphKernel()
1344 …tor<kernel::LiteKernel *> Scheduler::ScheduleSubGraphToSubGraphKernels(const int &subgraph_index) { in ScheduleSubGraphToSubGraphKernels()
1410 int Scheduler::ScheduleSubGraphToKernels(size_t subgraph_index, std::vector<kernel::LiteKernel *> *… in ScheduleSubGraphToKernels()
1703 auto &subgraph_index = pair.second; in SetSubgraphForPartialNode() local
|
| D | sub_graph_split.cc | 191 const schema::Primitive *SearchSubGraph::CreatePartialPrimitive(int64_t subgraph_index) { in CreatePartialPrimitive()
|
| /third_party/mindspore/mindspore/lite/tools/converter/legacy_optimizer/graph/ |
| D | infershape_pass.cc | 315 auto subgraph_index = PartialGraphIndex(partial_node); in CopyPartialShapeToSubGraph() local
346 auto subgraph_index = PartialGraphIndex(partial_node); in RestoreSubGraphInput() local
358 int subgraph_index = PartialGraphIndex(partial_node); in InferPartialNode() local
486 int InferShapePass::InferSubgraph(const int &subgraph_index, MetaGraphT *graph) { in InferSubgraph()
558 int InferShapePass::InitSearchTensor(const int &subgraph_index, MetaGraphT *graph, in InitSearchTensor()
|
| /third_party/mindspore/mindspore/lite/tools/converter/parser/tf/ |
| D | functionalize_control_op_pass.h | 59 static int subgraph_index = 1; in GetSubgraphIndex() local
|
| /third_party/mindspore/mindspore/lite/tools/anf_exporter/ |
| D | anf_exporter.cc | 251 const size_t &subgraph_index) { in SetSubGraphInputIndex()
278 int AnfExporter::SetSubGraphOutputIndex(const CNodePtr &cnode, const size_t subgraph_index, in SetSubGraphOutputIndex()
379 … const size_t &subgraph_index, const bool &keep_graph, const bool ©_primitive) { in Anf2Fb()
468 auto subgraph_index = meta_graphT->subGraph.size() - 1; in ExportSubgraph() local
|