Searched refs:sharding_colors_ (Results 1 – 1 of 1) sorted by relevance
417 sharding_colors_; member in xla::__anon7102e5d00111::HloDotDumper894 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()