Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_graph_dumper.cc412 absl::flat_hash_map<const HloComputation*, int64> cluster_ids_; member in xla::__anonf346b5ae0111::HloDotDumper
541 int64 cluster_id = cluster_ids_.at(from_node->parent()); in Header()
545 int64 cluster_id = cluster_ids_.at(to_node->parent()); in Header()
607 if (cluster_ids_.find(subcomp) != cluster_ids_.end()) { in DumpSubcomputation()
611 cluster_ids_[subcomp] = next_cluster_id_++; in DumpSubcomputation()