Searched defs:subgraph_kernel (Results 1 – 5 of 5) sorted by relevance
34 void set_subgraph_kernel(LiteKernel *subgraph_kernel) { subgraph_kernel_ = subgraph_kernel; } in set_subgraph_kernel()35 LiteKernel *subgraph_kernel() { return subgraph_kernel_; } in subgraph_kernel() function
226 auto *subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(kernel_); in CompileArrowThroughPartialCall() local497 int LiteSwitchOpActor::GetSwitchAndCallNode(kernel::SubGraphKernel *subgraph_kernel) { in GetSwitchAndCallNode()544 auto *subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(kernel_); in CompileArrowThroughSwitchCall() local
1248 auto subgraph_kernel = SchedulePartialToSubGraphKernel(subgraph_index); in SchedulePartialToKernel() local1335 auto subgraph_kernel = in SchedulePartialToSubGraphKernel() local1348 auto subgraph_kernel = SchedulePartialToSubGraphKernel(subgraph_index); in ScheduleSubGraphToSubGraphKernels() local1747 auto subgraph_kernel = in ConstructControlFlowMainGraph() local
205 auto *subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(kernel); in IsSwitchCall() local
245 auto subgraph_kernel = CreateNPUSubgraphKernel(connected_ops); in CreateSubgraphFromReadyOps() local