Searched refs:replicated_init (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.cc | 3113 auto replicated_init = GetPartitionedHlo(hlo->mutable_operand(1)) in HandleReduceWindow() local
3116 hlo->window(), hlo->sharding(), replicated_init.hlo()); in HandleReduceWindow()
3124 replicated_init.hlo()->shape(), in HandleReduceWindow()
3132 replicated_init.hlo(), resharded_operand_and_window->shard_window, in HandleReduceWindow()
3204 auto replicated_init = GetPartitionedHlo(hlo->mutable_operand(2)) in HandleSelectAndScatter() local
3289 hlo->sharding(), first_window_hlo, replicated_init.hlo(), in HandleSelectAndScatter()
3329 source_shard_hlo, replicated_init.hlo(), in HandleSelectAndScatter()
|