Searched refs:MultiplyAddDivideOffsetCalculation (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| H A D | spmd_partitioner_util.h | 223 class MultiplyAddDivideOffsetCalculation {
225 MultiplyAddDivideOffsetCalculation() in MultiplyAddDivideOffsetCalculation() function
227 MultiplyAddDivideOffsetCalculation(int64_t multiplier, int64_t offset,
231 const MultiplyAddDivideOffsetCalculation& other) const;
233 bool operator==(const MultiplyAddDivideOffsetCalculation& other) const {
261 const MultiplyAddDivideOffsetCalculation& copy_from) in OffsetCalculation()
265 const MultiplyAddDivideOffsetCalculation& lhs, in OffsetCalculation()
266 const MultiplyAddDivideOffsetCalculation& rhs) in OffsetCalculation()
296 MultiplyAddDivideOffsetCalculation copy_from_;
|
| H A D | spmd_partitioner_util.cc | 371 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in TileToPartialReplicateHaloExchange()
377 OffsetCalculation(MultiplyAddDivideOffsetCalculation(0, 0, 1)); in TileToPartialReplicateHaloExchange()
399 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in TileToPartialReplicateHaloExchange()
470 OffsetCalculation(MultiplyAddDivideOffsetCalculation(0, 0, 1)); in PadFromPartialReplicateShape()
475 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PadFromPartialReplicateShape()
512 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PadFromPartialReplicateShape()
565 MultiplyAddDivideOffsetCalculation::MultiplyAddDivideOffsetCalculation( in MultiplyAddDivideOffsetCalculation() function in xla::spmd::MultiplyAddDivideOffsetCalculation
572 OffsetCalculation MultiplyAddDivideOffsetCalculation::operator-( in operator -()
573 const MultiplyAddDivideOffsetCalculation& other) const { in operator -()
575 return OffsetCalculation(MultiplyAddDivideOffsetCalculation( in operator -()
[all …]
|
| H A D | convolution_handler.cc | 378 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
384 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
411 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
413 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
436 MultiplyAddDivideOffsetCalculation(0, new_padding_low_max, 1)) - in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
479 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
679 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
683 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
709 OffsetCalculation(MultiplyAddDivideOffsetCalculation()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
|
| H A D | fft_handler.cc | 63 OffsetCalculation(MultiplyAddDivideOffsetCalculation(0, 0, 1)); in PadEachPartitionWithHaloExchange()
71 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PadEachPartitionWithHaloExchange()
91 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PadEachPartitionWithHaloExchange()
|
| H A D | spmd_partitioner.cc | 675 std::vector<MultiplyAddDivideOffsetCalculation> start_on_padded_calculations( in ReshardAsWindowedInput()
677 std::vector<MultiplyAddDivideOffsetCalculation> limit_on_padded_calculations( in ReshardAsWindowedInput()
849 start_on_padded_calculations[i] = MultiplyAddDivideOffsetCalculation( in ReshardAsWindowedInput()
854 limit_on_padded_calculations[i] = MultiplyAddDivideOffsetCalculation( in ReshardAsWindowedInput()
922 MultiplyAddDivideOffsetCalculation( in ReshardAsWindowedInput()
926 MultiplyAddDivideOffsetCalculation(per_shard_window_counts[i], 0, in ReshardAsWindowedInput()
1095 MultiplyAddDivideOffsetCalculation shard_limit_of_previous_on_padded( in ReshardAsWindowedInput()
1101 MultiplyAddDivideOffsetCalculation shard_start_of_next_on_padded( in ReshardAsWindowedInput()
2347 MultiplyAddDivideOffsetCalculation( in HandleConcatenate()
2375 start_indices[dimension] = MultiplyAddDivideOffsetCalculation( in HandleConcatenate()
[all …]
|