Home
last modified time | relevance | path

Searched refs:cluster_from (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
357 int from = cluster_from->cycles_graph_node_id(); in MergeClusters()
368 cluster_from->Merge(cluster_to); in MergeClusters()
369 cluster_from->set_cycles_graph_node_id(*optional_merged_node); in MergeClusters()
380 Cluster* cluster_from = GetClusterForCyclesGraphNode(node); in ForEachEdgeInPostOrder() local
384 cycle_detector_.SuccessorsCopy(cluster_from->cycles_graph_node_id()); in ForEachEdgeInPostOrder()
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
394 int from = cluster_from->cycles_graph_node_id(); in MergeClusters()
399 VLOG(3) << "Could not contract " << cluster_from->DebugString(*graph_) in MergeClusters()
407 cluster_from->Merge(cluster_to); in MergeClusters()
409 cluster_from->set_cycles_graph_node_id(optional_merged_node.value()); in MergeClusters()
641 Cluster* cluster_from = GetClusterForCyclesGraphNode(node); in ForEachEdgeInPostOrder() local
642 if (!cluster_from) { in ForEachEdgeInPostOrder()
649 cycles_graph_.SuccessorsCopy(cluster_from->cycles_graph_node_id()); 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
[all …]