Searched refs:top_level_sharding_to_reset (Results 1 – 2 of 2) sorted by relevance
320 top_level_sharding_to_reset; in PartitionIndexParallelDimensions() local321 auto cleaner = MakeCleanup([&top_level_sharding_to_reset] { in PartitionIndexParallelDimensions()322 for (auto& to_reset : top_level_sharding_to_reset) { in PartitionIndexParallelDimensions()419 top_level_sharding_to_reset.emplace_back(operand.hlo(), in PartitionIndexParallelDimensions()467 top_level_sharding_to_reset; in PartitionGather() local468 auto cleaner = MakeCleanup([&top_level_sharding_to_reset] { in PartitionGather()469 for (auto& to_reset : top_level_sharding_to_reset) { in PartitionGather()
1663 top_level_sharding_to_reset; in PartitionDotGroupOnBatch() local1665 for (auto& to_reset : top_level_sharding_to_reset) { in PartitionDotGroupOnBatch()1729 top_level_sharding_to_reset.emplace_back(lhs.hlo(), lhs.sharding()); in PartitionDotGroupOnBatch()1731 top_level_sharding_to_reset.emplace_back(rhs.hlo(), rhs.sharding()); in PartitionDotGroupOnBatch()1818 top_level_sharding_to_reset.emplace_back(resharded.hlo(), in PartitionDotGroupOnBatch()1997 top_level_sharding_to_reset; in PartitionDotGroupOnNonContracting() local1999 for (auto& to_reset : top_level_sharding_to_reset) { in PartitionDotGroupOnNonContracting()2035 top_level_sharding_to_reset.emplace_back(matching.hlo(), matching.sharding()); in PartitionDotGroupOnNonContracting()2048 top_level_sharding_to_reset.emplace_back(other.hlo(), other.sharding()); in PartitionDotGroupOnNonContracting()2057 top_level_sharding_to_reset.emplace_back( in PartitionDotGroupOnNonContracting()[all …]