Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc3211 std::vector<MultiplyAddDivideOffsetCalculation> first_window( in HandleSelectAndScatter() local
3243 first_window[i] = MultiplyAddDivideOffsetCalculation( in HandleSelectAndScatter()
3258 first_window[i]; in HandleSelectAndScatter()
3267 HloOpcode::kMultiply, first_window[i], in HandleSelectAndScatter()
3280 (limit_window[i] - first_window[i]).MaxInRange(0, shard_count); in HandleSelectAndScatter()
3282 first_window[i].Calculate(partition_ordinals[i], &b_); in HandleSelectAndScatter()