• Home
  • Raw
  • Download

Lines Matching refs:subgraph

356   for (auto subgraph : *dst_kernels) {  in Schedule()
357 MS_LOG(DEBUG) << "[subgraph] : " << subgraph->name() << ", type:" << subgraph->subgraph_type(); in Schedule()
358 if (subgraph->desc().arch == kernel::KERNEL_ARCH::kDelegate) { in Schedule()
361 …kernel ::LiteKernel *> kernel_list = reinterpret_cast<kernel::SubGraphKernel *>(subgraph)->nodes(); in Schedule()
629 auto subgraph = src_model_->graph_.sub_graphs_.at(subgraph_index); in RestoreSubGraphInput() local
630 for (size_t i = 0; i < subgraph->input_indices_.size(); ++i) { in RestoreSubGraphInput()
631 auto &subgraph_input = src_tensors_->at(subgraph->input_indices_[i]); in RestoreSubGraphInput()
646 auto subgraph = src_model_->graph_.sub_graphs_.at(subgraph_index); in CopyPartialShapeToSubGraph() local
647 if (subgraph->input_indices_.size() != partial_node->input_indices_.size()) { in CopyPartialShapeToSubGraph()
650 << " subgraph input size: " << subgraph->input_indices_.size(); in CopyPartialShapeToSubGraph()
655 auto &subgraph_input = src_tensors_->at(subgraph->input_indices_[i]); in CopyPartialShapeToSubGraph()
730 auto subgraph = src_model_->graph_.sub_graphs_.at(subgraph_index); in InferSubGraphShape() local
732 for (auto node_index : subgraph->node_indices_) { in InferSubGraphShape()
1264 auto subgraph = src_model_->graph_.sub_graphs_.at(subgraph_index); in SubGraphPreferDataType() local
1265 for (auto node_index : subgraph->node_indices_) { in SubGraphPreferDataType()
1418 auto subgraph = src_model_->graph_.sub_graphs_.at(subgraph_index); in ScheduleSubGraphToKernels() local
1420 for (auto node_index : subgraph->node_indices_) { in ScheduleSubGraphToKernels()
1460 …std::transform(subgraph->input_indices_.begin(), subgraph->input_indices_.end(), std::back_inserte… in ScheduleSubGraphToKernels()
1464 …std::transform(subgraph->output_indices_.begin(), subgraph->output_indices_.end(), std::back_inser… in ScheduleSubGraphToKernels()
1564 auto subgraph = FindAllSubGraphKernels(src_kernel, *context_, &index, schema_version_); in ConstructSubGraphs() local
1565 if (subgraph == nullptr) { in ConstructSubGraphs()
1569 dst_kernel->emplace_back(subgraph); in ConstructSubGraphs()
1571 for (auto *subgraph : *dst_kernel) { in ConstructSubGraphs()
1573 if (subgraph->desc().arch != kernel::kDelegate) { in ConstructSubGraphs()
1575 auto ret = subgraph->Init(); in ConstructSubGraphs()