Searched refs:IsReplicated (Results 1 – 9 of 9) sorted by relevance
62 return !sharding.IsTileMaximal() || sharding.IsReplicated(); in IsSpatiallyPartitioned()536 if (lhs->sharding().IsReplicated()) { in InferConvolutionShardingFromOperands()611 if (op->has_sharding() && op->sharding().IsReplicated()) { in InferShardingFromOperands()702 if (operand->sharding().IsReplicated() || in InferShardingFromOperands()716 operand->sharding().IsReplicated() in InferShardingFromOperands()720 if (after_partial_replication.IsReplicated()) { in InferShardingFromOperands()743 if (!IsSpatiallyPartitioned(op) || op->sharding().IsReplicated()) { in InferShardingFromOperands()919 if (operand->sharding().IsReplicated()) { in InferShardingFromOperands()1136 if (user.sharding().IsReplicated()) { in GetShardingFromUser()1160 if (user.sharding().IsReplicated()) { in GetShardingFromUser()[all …]
126 bool IsReplicated() const { in IsReplicated() function131 tuple_elements_, [](const HloSharding& s) { return s.IsReplicated(); }); in IsReplicated()
303 *data = sharding.element(index).IsReplicated(); in ComputeHloReplicationOnComputation()344 sharding_tree.element(index).IsReplicated(); in ComputeHloReplication()
58 EXPECT_TRUE(sharding.IsReplicated()); in TEST_F()73 EXPECT_FALSE(sharding.IsReplicated()); in TEST_F()
59 } else if (!rhs.IsReplicated()) { in IsShardingMoreSpecific()65 return !lhs.IsReplicated(); in IsShardingMoreSpecific()1107 if (sharding.IsReplicated()) { in DevicesForShardingInternal()
552 if (IsReplicated()) { in ToProto()
1585 if (!output_sharding.IsReplicated()) { in PartitionBaseCase()1593 lhs.sharding().IsReplicated() && in PartitionBaseCase()1598 rhs.sharding().IsReplicated() && in PartitionBaseCase()1607 if (output_sharding.IsReplicated() && (should_partition_contracting_dim(0) || in PartitionBaseCase()1673 lhs.sharding().IsReplicated() in PartitionDotGroupOnBatch()1677 rhs.sharding().IsReplicated() in PartitionDotGroupOnBatch()1971 } else if (!other_sharding.IsReplicated()) { in GetNonContractingPartitionGroupedShardingForOtherOperand()2050 } else if (!other.sharding().IsReplicated()) { in PartitionDotGroupOnNonContracting()2759 if (lhs.sharding().IsReplicated() && rhs.sharding().IsReplicated() && in PartitionDot()
106 hlo->sharding().IsReplicated(); in ReportBeforePartition()442 if (!sharding().IsReplicated()) { in ReshardNoCache()485 if (sharding.IsReplicated() || EvenlyPartitions(base_shape_, sharding)) { in PadWithValue()739 if (sharding().IsReplicated()) { in ReshardAsWindowedInput()827 if (entry.first.IsReplicated()) { in Replicate()836 if (sharding.IsReplicated()) { in Replicate()840 if (entry.first.IsReplicated()) { in Replicate()1811 if (sharding.IsTileMaximal() || sharding.IsReplicated()) { in HandleCustomCall()3048 if (hlo->sharding().IsReplicated()) { in HandleRng()3784 TF_RET_CHECK(root_sharding.IsReplicated() || in PreprocessSharding()[all …]
51 return sharding.IsReplicated(); in HasReplicatedSharding()138 return sharding.IsReplicated(); in EvenlyPartitions()179 if (sharding.IsReplicated()) { in MakeNonPaddedShapeForGivenPartition()1801 if (source_sharding.IsReplicated()) { in CreateMatchingShardingOnDims()1853 if (new_index_shard.IsReplicated()) { in GatherOperandsShardedAcrossParallelDims()1860 if (new_operand_shard.IsReplicated()) { in GatherOperandsShardedAcrossParallelDims()