Searched defs:input_kernels (Results 1 – 3 of 3) sorted by relevance
60 auto input_kernels = kernel::LiteKernelUtil::SubgraphInputNodes(kernels); in TEST_F() local
88 const std::set<std::shared_ptr<KernelDef>> &input_kernels() { return input_kernels_; } in input_kernels() function
67 kernel::SubGraphKernel *CreateCustomSubGraph(std::vector<kernel::LiteKernel *> &&input_kernels, in CreateCustomSubGraph()1132 …std::vector<kernel::LiteKernel *> input_kernels = kernel::LiteKernelUtil::SubgraphInputNodes(kerne… in CreateSubGraphKernel() local