Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc3314 Window window_on_shard = hlo->window(); in HandleSelectAndScatter() local
3315 for (int64 i = 0; i < window_on_shard.dimensions_size(); ++i) { in HandleSelectAndScatter()
3320 auto reshard_wd = window_on_shard.mutable_dimensions(i); in HandleSelectAndScatter()
3328 data_shard_hlo->shape(), data_shard_hlo, select, window_on_shard, in HandleSelectAndScatter()
3340 for (int64 i = 0; i < window_on_shard.dimensions_size(); ++i) { in HandleSelectAndScatter()