Home
last modified time | relevance | path

Searched defs:top_level_sharding_to_reset (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc320 top_level_sharding_to_reset; in PartitionIndexParallelDimensions() local
467 top_level_sharding_to_reset; in PartitionGather() local
Ddot_handler.cc1663 top_level_sharding_to_reset; in PartitionDotGroupOnBatch() local
1997 top_level_sharding_to_reset; in PartitionDotGroupOnNonContracting() local
2092 top_level_sharding_to_reset; in PartitionDotGroupOnContracting() local