Searched refs:IsCollective (Results 1 – 10 of 10) sorted by relevance
41 node->IsCollective() && node->type_string() == "CollectiveReduce"; in DiscoverDataDependencies()78 if (!collective_nodes[i]->IsCollective() || in CreateControlDependencies()
41 if (!src->IsCollective()) { in VerifyGraph()68 if (node->IsCollective() || in VerifyAttrs()
182 bool IsCollective() const { return class_ == NC_COLLECTIVE; } in IsCollective() function
43 IsCollective(node) || in IsDenylisted()
1184 if (!IsCollective(n)) return false; in IsCollectiveNode()
54 bool IsCollective(const NodeDef& node);
141 bool IsCollective(const NodeDef& node) { in IsCollective() function
395 if (n->IsCollective()) { in InferAllocAttr()
910 if (node->IsCollective()) { in BuildGraph()
93 if (n->IsCollective()) { in Initialize()