Searched defs:cross_partition_spmd (Results 1 – 2 of 2) sorted by relevance
46 bool cross_partition_spmd, in DetermineHloInstructionIsReplicated()486 bool cross_partition_spmd) { in Run()492 HloReplicationAnalysis::Run(const HloModule* module, bool cross_partition_spmd, in Run()504 bool cross_partition_spmd) { in RunWithPartialReplication()
107 HloReplicationAnalysis(const HloModule* module, bool cross_partition_spmd, in HloReplicationAnalysis()