Searched refs:sharding_colors_ (Results 1 – 1 of 1) sorted by relevance
423 sharding_colors_; member in xla::__anonf346b5ae0111::HloDotDumper909 auto it = sharding_colors_.find(instr->sharding()); in GetInstructionColor()910 if (it != sharding_colors_.end()) { in GetInstructionColor()915 sharding_colors_.emplace(instr->sharding(), color); in GetInstructionColor()