Searched refs:kPrimGetNext (Results 1 – 4 of 4) sorted by relevance
36 GetNext() : PrimitiveC(prim::kPrimGetNext->name()) {} in GetNext()
93 REGISTER_PRIMITIVE_EVAL_IMPL(GetNext, prim::kPrimGetNext, GetNextInfer, nullptr, true);
372 … if (node_name == prim::kPrimGetNext->name() || node_name == prim::kPrimInitDataSetQueue->name()) { in isDatasetGraph()
107 inline const PrimitivePtr kPrimGetNext = std::make_shared<Primitive>(kGetNext); variable