Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_graph_dumper.cc417 sharding_colors_; member in xla::__anon7102e5d00111::HloDotDumper
894 auto it = sharding_colors_.find(instr->sharding()); in GetInstructionColor()
895 if (it != sharding_colors_.end()) { in GetInstructionColor()
900 sharding_colors_.emplace(instr->sharding(), color); in GetInstructionColor()