Searched defs:partial_kernel (Results 1 – 2 of 2) sorted by relevance
77 auto partial_kernel = reinterpret_cast<kernel::PartialFusionKernel *>(partial_node->kernel()); in SplitNonTailCallSubGraphs() local488 auto partial_kernel = reinterpret_cast<kernel::PartialFusionKernel *>(partial_node->kernel()); in BuildBoundaryForMultipleCalledGraph() local530 auto partial_kernel = reinterpret_cast<kernel::PartialFusionKernel *>(partial->kernel()); in GetTailCallFinalSubgraphs() local721 auto partial_kernel = reinterpret_cast<kernel::PartialFusionKernel *>(partial_node->kernel()); in IsolateInputOfMultipleCalledGraph() local750 auto partial_kernel = static_cast<kernel::PartialFusionKernel *>((pair.first)->kernel()); in SetSubgraphForPartialNode() local777 auto partial_kernel = reinterpret_cast<kernel::PartialFusionKernel *>((pair.first)->kernel()); in RecordPartialInputLinkInfo() local
617 auto partial_kernel = reinterpret_cast<PartialFusionKernel *>(partial_node->kernel()); in GetCallInputPartialsCorrespondingOutputSubgraph() local