Searched refs:kPrimInitDataSetQueue (Results 1 – 4 of 4) sorted by relevance
372 … if (node_name == prim::kPrimGetNext->name() || node_name == prim::kPrimInitDataSetQueue->name()) { in isDatasetGraph()
432 inline const PrimitivePtr kPrimInitDataSetQueue = std::make_shared<Primitive>("InitDataSetQueue"); variable
1393 if (node_name == prim::kPrimInitDataSetQueue->name()) { in IsDatasetGraph()
411 if (AnfAlgo::CheckPrimitiveType(output_with_index.first, prim::kPrimInitDataSetQueue)) { in GetAllOutputWithIndex()