Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dgather_scatter_handler.cc322 for (auto& to_reset : top_level_sharding_to_reset) { in PartitionIndexParallelDimensions() local
469 for (auto& to_reset : top_level_sharding_to_reset) { in PartitionGather() local
Ddot_handler.cc1665 for (auto& to_reset : top_level_sharding_to_reset) { in PartitionDotGroupOnBatch() local
1999 for (auto& to_reset : top_level_sharding_to_reset) { in PartitionDotGroupOnNonContracting() local
2094 for (auto& to_reset : top_level_sharding_to_reset) { in PartitionDotGroupOnContracting() local