Searched defs:kernel_arg (Results 1 – 4 of 4) sorted by relevance
89 const IntArrayRef kernel_arg, in pool2d()230 const IntArrayRef kernel_arg, in avg_pool2d()248 const IntArrayRef kernel_arg, in max_pool2d()
59 } kernel_arg; typedef
412 for (auto kernel_arg : kernel_thunk->arguments()) in Rewrite() local
133 static std::string generate_argument(const KernelArgInfo& kernel_arg) in generate_argument()