Searched defs:input_kernels (Results 1 – 3 of 3) sorted by relevance
57 auto input_kernels = kernel::KernelExecUtil::SubgraphInputNodes(kernels); in TEST_F() local
89 const std::set<std::shared_ptr<KernelDef>> &input_kernels() const { return input_kernels_; } in input_kernels() function
355 SubGraphKernel *CreateCustomSubGraph(std::vector<KernelExec *> &&input_kernels, in CreateCustomSubGraph()388 std::vector<KernelExec *> input_kernels = SubgraphInputNodes(kernels); in CreateSubGraphKernel() local