Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner.cc572 std::vector<MultiplyAddDivideOffsetCalculation> limit_on_padded_calculations( in ReshardAsWindowedInput() local
627 limit_on_padded_calculations[i] = MultiplyAddDivideOffsetCalculation( in ReshardAsWindowedInput()
636 limit_on_padded_calculations[i] - start_on_padded_calculations[i]; in ReshardAsWindowedInput()
640 i, limit_on_padded_calculations[i].Calculate(shard_count - 1)); in ReshardAsWindowedInput()
802 limit_on_padded_calculations[dim] - shard_start_of_next_on_padded; in ReshardAsWindowedInput()