Searched defs:subgraph_kernel (Results 1 – 11 of 11) sorted by relevance
49 auto subgraph_kernel = dynamic_cast<kernel::SubGraphKernel *>(kernel); in PrepareKernels() local72 bool ExecutionPlan::CalcTensorRefCount(abstract::Kernel *subgraph_kernel) { in CalcTensorRefCount()
40 auto subgraph_kernel = in ConstructFusionKernel() local
61 auto subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(cur); in SplitNonTailCallSubGraphs() local107 auto subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(kernel); in GetNonTailCallSubGraphs() local150 int ControlFlowScheduler::SplitSubGraphNodesIntoTwoParts(kernel::SubGraphKernel *subgraph_kernel, in SplitSubGraphNodesIntoTwoParts()189 int ControlFlowScheduler::SplitSingleNonTailCallSubGraph(kernel::SubGraphKernel *subgraph_kernel, in SplitSingleNonTailCallSubGraph()219 auto subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(kernel); in RemoveUselessKernels() local280 auto subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(dst_kernel); in RecordAllNonTailCallLinkInfo() local535 auto subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(subgraph); in GetTailCallFinalSubgraphs() local578 auto subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(dst_kernel); in RecordAllTailCallLinkInfo() local781 auto subgraph_kernel = subgraph_kernels.front(); in RecordPartialInputLinkInfo() local
35 auto subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(kernel); in CreateActor() local
47 auto subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(kernel_); in RunOpData() local103 auto subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(kernel_); in KernelActorInit() local
347 bool KernelExecUtil::IsOutputSubGraph(const KernelExec *subgraph_kernel) { in IsOutputSubGraph()445 auto subgraph_kernel = reinterpret_cast<SubGraphKernel *>(kernel); in ReplaceSubGraphNodesInTensor() local472 auto subgraph_kernel = reinterpret_cast<SubGraphKernel *>(kernel); in ReplaceSubGraphNodesOutTensor() local516 auto *subgraph_kernel = reinterpret_cast<SubGraphKernel *>(kernel); in IsSwitchTypeCall() local550 auto subgraph_kernel = reinterpret_cast<SubGraphKernel *>(kernel); in IsNonTailCallSubGraph() local560 auto subgraph_kernel = reinterpret_cast<SubGraphKernel *>(kernel); in IsTailCallSubGraph() local
204 auto *subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(kernel_); in UpdateActorOutput() local400 auto subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(kernel_); in InitInputData() local408 auto subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(kernel_); in InitInputData() local
1331 auto subgraph_kernel = SchedulePartialToSubGraphKernel(subgraph_index); in SchedulePartialToKernel() local1421 auto subgraph_kernel = kernel::KernelExecUtil::CreateSubGraphKernel( in SchedulePartialToSubGraphKernel() local1434 auto subgraph_kernel = SchedulePartialToSubGraphKernel(subgraph_index); in ScheduleSubGraphToSubGraphKernels() local1694 auto subgraph_kernel = static_cast<kernel::SubGraphKernel *>(subgraph); in ConstructNormalSubGraphs() local1799 auto subgraph_kernel = kernel::KernelExecUtil::CreateSubGraphKernel( in ConstructControlFlowMainGraph() local
813 auto subgraph_kernel = static_cast<kernel::SubGraphKernel *>(kernel); in PrepareKernels() local833 auto subgraph_kernel = static_cast<kernel::SubGraphKernel *>(kernel); in PrepareKernels() local
65 int LiteSwitchOpActor::GetSwitchAndCallNode(kernel::SubGraphKernel *subgraph_kernel) { in GetSwitchAndCallNode()98 auto *subgraph_kernel = reinterpret_cast<kernel::SubGraphKernel *>(kernel_); in ModifySubgraphKernel() local
245 auto subgraph_kernel = CreateNPUSubgraphKernel(connected_ops); in CreateSubgraphFromReadyOps() local