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
483 top_level_sharding_to_reset; in PartitionGather() local
Ddot_handler.cc1752 top_level_sharding_to_reset; in PartitionDotGroupOnBatch() local
2088 top_level_sharding_to_reset; in PartitionDotGroupOnNonContracting() local
2276 top_level_sharding_to_reset; in PartitionDotGroupOnContracting() local