Searched defs:output_kernels (Results 1 – 4 of 4) sorted by relevance
59 auto output_kernels = kernel::KernelExecUtil::SubgraphOutputNodes(kernels); in TEST_F() local
356 std::vector<KernelExec *> &&output_kernels, in CreateCustomSubGraph()389 std::vector<KernelExec *> output_kernels = SubgraphOutputNodes(kernels); in CreateSubGraphKernel() local
209 auto output_kernels = subgraph_kernel->out_nodes(); in UpdateActorOutput() local
624 auto output_kernels = partial_in_kernel->out_kernels(); in IsolatePartialInputs() local