Searched defs:CanSideEffectingHaveReplicatedSharding (Results 1 – 3 of 3) sorted by relevance
73 bool GpuSpmdPartitioner::CanSideEffectingHaveReplicatedSharding( in CanSideEffectingHaveReplicatedSharding() function in xla::gpu::GpuSpmdPartitioner
75 bool StatefulRngSpmdPartitioner::CanSideEffectingHaveReplicatedSharding( in CanSideEffectingHaveReplicatedSharding() function in xla::spmd::StatefulRngSpmdPartitioner
272 virtual bool CanSideEffectingHaveReplicatedSharding( in CanSideEffectingHaveReplicatedSharding() function