Searched refs:kPrimTopK (Results 1 – 2 of 2) sorted by relevance
168 inline const PrimitivePtr kPrimTopK = std::make_shared<Primitive>("TopK"); variable
1727 if (name == prim::kPrimTopK->name()) { in CheckCNode()