Searched defs:kernel_select (Results 1 – 3 of 3) sorted by relevance
42 … const KernelSelectPtr &kernel_select, const std::vector<size_t> &dst_shape) { in CreateReshapeNode()109 const KernelSelectPtr &kernel_select) { in GetTransInputNodePtr()140 const KernelSelectPtr &kernel_select) { in InsertTransOpForSingleOutput()157 … const AnfNodePtr &node, const KernelSelectPtr &kernel_select) { in InsertTransOpForMultipleOutput()193 … const KernelSelectPtr &kernel_select, size_t insert_index, bool is_insert_input) { in AddTransOpNodeToGraph()262 …Node(const FuncGraphPtr &func_graph, const AnfNodePtr &input, const KernelSelectPtr &kernel_select, in NewTransOpNode()369 const KernelSelectPtr &kernel_select) { in InsertTransOpForOutput()388 const KernelSelectPtr &kernel_select) { in InsertTransOpForInput()
56 KernelSelectPtr kernel_select = std::make_shared<KernelSelect>(); in Insert() local
122 auto kernel_select = std::make_shared<KernelSelect>(); in AddAdditionalToRefOutput() local