Searched refs:instruction_sharding (Results 1 – 1 of 1) sorted by relevance
477 auto instruction_sharding = instruction->sharding_ptr(); in operator ()() local480 if (instruction_sharding == nullptr && root_sharding == nullptr) { in operator ()()484 if (instruction_sharding != nullptr && root_sharding != nullptr && in operator ()()485 ShardingMatches(*instruction_sharding, *root_sharding)) { in operator ()()489 if (instruction_sharding != nullptr) { in operator ()()490 instruction_sharding = CloneShardingForDomain(instruction_sharding); in operator ()()496 auto it = domain_cse_map_.find({operand, instruction_sharding}); in operator ()()505 << (instruction_sharding != nullptr ? instruction_sharding->ToString() in operator ()()514 std::make_unique<ShardingMetadata>(instruction_sharding))); in operator ()()515 domain_cse_map_.emplace(DomainCseMapKey{operand, instruction_sharding}, in operator ()()