Searched refs:instruction_sharding (Results 1 – 1 of 1) sorted by relevance
476 auto instruction_sharding = instruction->sharding_ptr(); in operator ()() local479 if (instruction_sharding == nullptr && root_sharding == nullptr) { in operator ()()483 if (instruction_sharding != nullptr && root_sharding != nullptr && in operator ()()484 ShardingMatches(*instruction_sharding, *root_sharding)) { in operator ()()488 if (instruction_sharding != nullptr) { in operator ()()489 instruction_sharding = CloneShardingForDomain(instruction_sharding); in operator ()()495 auto it = domain_cse_map_.find({operand, instruction_sharding}); in operator ()()504 << (instruction_sharding != nullptr ? instruction_sharding->ToString() in operator ()()513 absl::make_unique<ShardingMetadata>(instruction_sharding))); in operator ()()514 domain_cse_map_.emplace(DomainCseMapKey{operand, instruction_sharding}, in operator ()()