Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc1221 auto remaining_source_target_dims = source_target_dims; in ReshardWithAllToAll() local
1222 remaining_source_target_dims.remove_prefix(1); in ReshardWithAllToAll()
1224 .ReshardWithAllToAll(target, remaining_source_target_dims); in ReshardWithAllToAll()