Searched refs:start_offset_on_padded_concat_calculation (Results 1 – 2 of 2) sorted by relevance
398 OffsetCalculation start_offset_on_padded_concat_calculation = in TileToPartialReplicateHaloExchange() local409 slice_offsets[dim] = start_offset_on_padded_concat_calculation.Calculate( in TileToPartialReplicateHaloExchange()511 OffsetCalculation start_offset_on_padded_concat_calculation = in PadFromPartialReplicateShape() local518 slice_offsets[dim] = start_offset_on_padded_concat_calculation.Calculate( in PadFromPartialReplicateShape()913 auto start_offset_on_padded_concat_calculation = in ExchangeHaloAndGetValidData() local923 start_offset_on_padded_concat_calculation.MaxInRange(0, shard_count) + in ExchangeHaloAndGetValidData()967 slice_offsets[dim] = start_offset_on_padded_concat_calculation.Calculate( in ExchangeHaloAndGetValidData()
90 OffsetCalculation start_offset_on_padded_concat_calculation = in PadEachPartitionWithHaloExchange() local102 start_offset_on_padded_concat_calculation.Calculate( in PadEachPartitionWithHaloExchange()