Home
last modified time | relevance | path

Searched refs:CanReshardWithCollectivePermute (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h291 bool CanReshardWithCollectivePermute(const HloSharding& source,
Dspmd_partitioner.cc409 if (CanReshardWithCollectivePermute(sharding(), target)) { in ReshardNoCache()
926 if (CanReshardWithCollectivePermute(sharding(), temp_sharding)) { in ReshardToPartialReplicateWithAllGather()
1047 if (CanReshardWithCollectivePermute(temp_target_sharding, target)) { in ReshardFromPartialReplicateWithDynamicSlice()
1315 CHECK(CanReshardWithCollectivePermute(sharding(), target)) in ReshardWithCollectivePermute()
Ddot_handler.cc543 !CanReshardWithCollectivePermute( in PartitionBaseCase()
560 !CanReshardWithCollectivePermute( in PartitionBaseCase()
Dspmd_partitioner_util.cc1369 bool CanReshardWithCollectivePermute(const HloSharding& source, in CanReshardWithCollectivePermute() function