Searched refs:start_offset_on_padded_concat_calculation (Results 1 – 2 of 2) sorted by relevance
92 OffsetCalculation start_offset_on_padded_concat_calculation = in PadEachPartitionWithHaloExchange() local104 start_offset_on_padded_concat_calculation.Calculate( in PadEachPartitionWithHaloExchange()
483 OffsetCalculation start_offset_on_padded_concat_calculation = in TileToPartialReplicateHaloExchange() local494 slice_offsets[dim] = start_offset_on_padded_concat_calculation.Calculate( in TileToPartialReplicateHaloExchange()595 OffsetCalculation start_offset_on_padded_concat_calculation = in PadFromPartialReplicateShape() local602 slice_offsets[dim] = start_offset_on_padded_concat_calculation.Calculate( in PadFromPartialReplicateShape()951 auto start_offset_on_padded_concat_calculation = in ExchangeHaloAndGetValidData() local961 start_offset_on_padded_concat_calculation.MaxInRange(0, shard_count) + in ExchangeHaloAndGetValidData()1005 slice_offsets[dim] = start_offset_on_padded_concat_calculation.Calculate( in ExchangeHaloAndGetValidData()