Searched defs:cluster_nodes (Results 1 – 2 of 2) sorted by relevance
195 absl::flat_hash_map<Node*, int>* cluster_nodes, in ReplicateNodesAndEdges()278 absl::flat_hash_map<Node*, int>& cluster_nodes = it.second; in ReplicatePerReplicaNodesInFunctionGraph() local
52 const std::vector<const NodeDef*>& cluster_nodes, in FilterPartitionedConstNodes()