Searched refs:kPrimAllToAllv (Results 1 – 3 of 3) sorted by relevance
50 if (IsPrimitiveCNode(n, prim::kPrimAllToAllv)) { in TEST_F()70 if (IsPrimitiveCNode(n, prim::kPrimAllToAllv)) { in TEST_F()
128 if (!IsPrimitiveCNode(cnode, prim::kPrimAllToAllv)) { in SetAllToAllvAttr()158 } else if (IsPrimitiveCNode(cnode, prim::kPrimAllToAllv)) { in GetGeNodeName()
423 inline const PrimitivePtr kPrimAllToAllv = std::make_shared<Primitive>("AllToAllv"); variable