Searched refs:kSplitOpName (Results 1 – 6 of 6) sorted by relevance
71 Register(kSplitOpName, {0}); in ConstInputToAttrInfoRegistry()
158 …if ((op_name == kSplitOpName || op_name == kSplitVOpName) && AnfAlgo::HasNodeAttr(kAttrNonTask, an… in LaunchKernel()191 if ((AnfAlgo::GetCNodeName(prenode_with_index.first) == kSplitOpName || in LaunchKernel()
31 const std::set<std::string> InvalidOps = {kSplitOpName, kSplitVOpName, kConcatOpName};
188 auto split_prim = std::make_shared<Primitive>(kSplitOpName); in DefinePattern()
100 constexpr auto kSplitOpName = "Split"; variable
849 …if ((op_name == kSplitOpName || op_name == kSplitVOpName) && AnfAlgo::HasNodeAttr(kAttrNonTask, ke… in NonTaskSplitProcess()