• Home
  • Raw
  • Download

Lines Matching refs:kernels

69 …                        const std::vector<kernel::LiteKernel *> &kernels, kernel::Kernel *kernel) {  in CreateCustomSubGraph()  argument
70 …kernel = new (std::nothrow) kernel::CustomSubGraph(input_kernels, output_kernels, kernels, kernel); in CreateCustomSubGraph()
388 std::vector<kernel::Kernel *> kernels; in ReplaceDelegateKernels() local
390 kernels.push_back((*dst_kernels)[i]->kernel()); in ReplaceDelegateKernels()
397 …new (std::nothrow) DelegateModel<schema::Primitive>(&kernels, ms_inputs_, ms_outputs_, primitives_… in ReplaceDelegateKernels()
421 for (auto kernel : kernels) { in ReplaceDelegateKernels()
1109 kernel::SubGraphKernel *CreateSubGraphKernel(const std::vector<kernel::LiteKernel *> &kernels, in CreateSubGraphKernel() argument
1121 input_tensors = kernel::LiteKernelUtil::SubgraphInputTensors(kernels); in CreateSubGraphKernel()
1126 output_tensors = kernel::LiteKernelUtil::SubgraphOutputTensors(kernels); in CreateSubGraphKernel()
1132 …::vector<kernel::LiteKernel *> input_kernels = kernel::LiteKernelUtil::SubgraphInputNodes(kernels); in CreateSubGraphKernel()
1133 …vector<kernel::LiteKernel *> output_kernels = kernel::LiteKernelUtil::SubgraphOutputNodes(kernels); in CreateSubGraphKernel()
1136 …sub_graph = CreateCustomSubGraph(std::move(input_kernels), std::move(output_kernels), kernels, inn… in CreateSubGraphKernel()
1140 …sub_graph = new (std::nothrow) kernel::OpenCLSubGraph(input_kernels, output_kernels, kernels, inne… in CreateSubGraphKernel()
1153 …sub_graph = new (std::nothrow) kernel::CpuFp16SubGraph(input_kernels, output_kernels, kernels, inn… in CreateSubGraphKernel()
1171 …sub_graph = new (std::nothrow) kernel::CpuFp32SubGraph(input_kernels, output_kernels, kernels, inn… in CreateSubGraphKernel()
1302 std::vector<kernel::LiteKernel *> kernels; in ScheduleMainSubGraphToKernels() local
1305 auto ret = ScheduleSubGraphToKernels(kMainSubGraphIndex, &kernels, &in_tensors, &out_tensors); in ScheduleMainSubGraphToKernels()
1310 return kernels; in ScheduleMainSubGraphToKernels()
1321 std::vector<kernel::LiteKernel *> kernels; in SchedulePartialToSubGraphKernel() local
1324 …auto ret = ScheduleSubGraphToKernels(subgraph_index, &kernels, &in_tensors, &out_tensors, prefer_d… in SchedulePartialToSubGraphKernel()
1329 kernel::LiteKernelUtil::FindAllInoutKernels(kernels); in SchedulePartialToSubGraphKernel()
1331 if (!kernels.empty()) { in SchedulePartialToSubGraphKernel()
1332 cur_sub_graph_type = GetKernelSubGraphType(kernels.front(), *context_, true); in SchedulePartialToSubGraphKernel()
1336 …CreateSubGraphKernel(kernels, &in_tensors, &out_tensors, cur_sub_graph_type, *context_, schema_ver… in SchedulePartialToSubGraphKernel()
1400 auto kernels = ScheduleSubGraphToSubGraphKernels(cur_subgraph_index); in ScheduleGraphToKernels() local
1401 if (kernels.empty()) { in ScheduleGraphToKernels()
1405 std::copy(kernels.begin(), kernels.end(), std::back_inserter(*dst_kernels)); in ScheduleGraphToKernels()
1638 …el::SubGraphType Scheduler::PartialSubGraphType(const std::vector<kernel::LiteKernel *> &kernels) { in PartialSubGraphType() argument
1639 if (std::any_of(kernels.begin(), kernels.end(), in PartialSubGraphType()
1723 bool Scheduler::IsControlFlowParttern(const std::vector<kernel::LiteKernel *> &kernels) { in IsControlFlowParttern() argument
1724 if (std::any_of(kernels.begin(), kernels.end(), [](kernel::LiteKernel *item) { in IsControlFlowParttern()
1735 int Scheduler::ConstructControlFlowMainGraph(std::vector<kernel::LiteKernel *> *kernels) { in ConstructControlFlowMainGraph() argument
1736 auto back_kernels = *kernels; in ConstructControlFlowMainGraph()
1737 kernels->clear(); in ConstructControlFlowMainGraph()
1741 kernels->push_back(kernel); in ConstructControlFlowMainGraph()
1753 kernels->insert(kernels->begin(), subgraph_kernel); in ConstructControlFlowMainGraph()