Home
last modified time | relevance | path

Searched refs:cluster_to (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/
Dmhlo_fusion.cc356 bool MergeClusters(Cluster* cluster_from, Cluster* cluster_to) { in MergeClusters() argument
358 int to = cluster_to->cycles_graph_node_id(); in MergeClusters()
368 cluster_from->Merge(cluster_to); in MergeClusters()
387 Cluster* cluster_to = GetClusterForCyclesGraphNode(to); in ForEachEdgeInPostOrder() local
388 bool contracted_edge = fn(cluster_from, cluster_to); in ForEachEdgeInPostOrder()
/external/tensorflow/tensorflow/compiler/jit/
Dmark_for_compilation_pass.cc393 bool MergeClusters(Cluster* cluster_from, Cluster* cluster_to) { in MergeClusters() argument
395 int to = cluster_to->cycles_graph_node_id(); in MergeClusters()
400 << " -> " << cluster_to->DebugString(*graph_) in MergeClusters()
407 cluster_from->Merge(cluster_to); in MergeClusters()
654 Cluster* cluster_to = GetClusterForCyclesGraphNode(to); in ForEachEdgeInPostOrder() local
655 if (!cluster_to) { in ForEachEdgeInPostOrder()
659 TF_ASSIGN_OR_RETURN(bool contracted_edge, fn(cluster_from, cluster_to)); in ForEachEdgeInPostOrder()
909 const Cluster& cluster_from, const Cluster& cluster_to) { in ClusteringWillIntroduceInterDeviceDependency() argument
921 cycles_graph_.Predecessors(cluster_to.cycles_graph_node_id())) { in ClusteringWillIntroduceInterDeviceDependency()
925 AreDevicesCompatible(cluster_to, *cluster_in)); in ClusteringWillIntroduceInterDeviceDependency()