Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h287 GetReshardAllToAllSourceTargetDims(const HloSharding& source,
Dspmd_partitioner.cc414 GetReshardAllToAllSourceTargetDims(sharding(), target)) { in ReshardNoCache()
1295 if (auto src_tgt_dims = GetReshardAllToAllSourceTargetDims( in ReshardPartialReplicateWithAllToAll()
1304 if (auto src_tgt_dims = GetReshardAllToAllSourceTargetDims( in ReshardPartialReplicateWithAllToAll()
Ddot_handler.cc545 !GetReshardAllToAllSourceTargetDims( in PartitionBaseCase()
562 !GetReshardAllToAllSourceTargetDims( in PartitionBaseCase()
Dspmd_partitioner_util.cc1284 GetReshardAllToAllSourceTargetDims(const HloSharding& source, in GetReshardAllToAllSourceTargetDims() function