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()
172 bool IsCollective() const { return class_ == NC_COLLECTIVE; } in IsCollective() function
42 IsCollective(node) || in IsBlacklisted()
913 if (!IsCollective(n)) return false; in IsCollectiveNode()
47 bool IsCollective(const NodeDef& node);
111 bool IsCollective(const NodeDef& node) { in IsCollective() function
85 if (n->IsCollective()) { in Initialize()
798 if (node->IsCollective()) { in BuildGraph()
840 if (n->IsCollective()) { in InferAllocAttr()