Searched refs:limit_window (Results 1 – 1 of 1) sorted by relevance
3214 std::vector<MultiplyAddDivideOffsetCalculation> limit_window( in HandleSelectAndScatter() local3250 limit_window[i] = MultiplyAddDivideOffsetCalculation( in HandleSelectAndScatter()3260 limit_window[i] - MultiplyAddDivideOffsetCalculation( in HandleSelectAndScatter()3271 HloOpcode::kMultiply, limit_window[i], in HandleSelectAndScatter()3280 (limit_window[i] - first_window[i]).MaxInRange(0, shard_count); in HandleSelectAndScatter()3288 limit_window[i].Calculate(shard_count - 1), max_windows, i, in HandleSelectAndScatter()3300 (limit_window[i].Calculate(shard_count - 1) - 1) * wd.stride() + in HandleSelectAndScatter()