Searched refs:MultiplyAddDivideOffsetCalculation (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner_util.h | 138 class MultiplyAddDivideOffsetCalculation {
140 MultiplyAddDivideOffsetCalculation() in MultiplyAddDivideOffsetCalculation() function
142 MultiplyAddDivideOffsetCalculation(int64 multiplier, int64 offset,
146 const MultiplyAddDivideOffsetCalculation& other) const;
148 bool operator==(const MultiplyAddDivideOffsetCalculation& other) const {
176 const MultiplyAddDivideOffsetCalculation& copy_from) in OffsetCalculation()
180 const MultiplyAddDivideOffsetCalculation& lhs, in OffsetCalculation()
181 const MultiplyAddDivideOffsetCalculation& rhs) in OffsetCalculation()
211 MultiplyAddDivideOffsetCalculation copy_from_;
|
| D | convolution_handler.cc | 380 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
386 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
413 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
415 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
438 MultiplyAddDivideOffsetCalculation(0, new_padding_low_max, 1)) - in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
479 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
680 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
684 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
710 OffsetCalculation(MultiplyAddDivideOffsetCalculation()); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
|
| D | spmd_partitioner_util.cc | 456 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in TileToPartialReplicateHaloExchange()
462 OffsetCalculation(MultiplyAddDivideOffsetCalculation(0, 0, 1)); in TileToPartialReplicateHaloExchange()
484 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in TileToPartialReplicateHaloExchange()
555 OffsetCalculation(MultiplyAddDivideOffsetCalculation(0, 0, 1)); in PadFromPartialReplicateShape()
560 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PadFromPartialReplicateShape()
596 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PadFromPartialReplicateShape()
649 MultiplyAddDivideOffsetCalculation::MultiplyAddDivideOffsetCalculation( in MultiplyAddDivideOffsetCalculation() function in xla::spmd::MultiplyAddDivideOffsetCalculation
656 OffsetCalculation MultiplyAddDivideOffsetCalculation::operator-( in operator -()
657 const MultiplyAddDivideOffsetCalculation& other) const { in operator -()
659 return OffsetCalculation(MultiplyAddDivideOffsetCalculation( in operator -()
[all …]
|
| D | fft_handler.cc | 65 OffsetCalculation(MultiplyAddDivideOffsetCalculation(0, 0, 1)); in PadEachPartitionWithHaloExchange()
73 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PadEachPartitionWithHaloExchange()
93 OffsetCalculation(MultiplyAddDivideOffsetCalculation( in PadEachPartitionWithHaloExchange()
|
| D | spmd_partitioner.cc | 570 std::vector<MultiplyAddDivideOffsetCalculation> start_on_padded_calculations( in ReshardAsWindowedInput()
572 std::vector<MultiplyAddDivideOffsetCalculation> limit_on_padded_calculations( in ReshardAsWindowedInput()
622 start_on_padded_calculations[i] = MultiplyAddDivideOffsetCalculation( in ReshardAsWindowedInput()
627 limit_on_padded_calculations[i] = MultiplyAddDivideOffsetCalculation( in ReshardAsWindowedInput()
694 MultiplyAddDivideOffsetCalculation( in ReshardAsWindowedInput()
698 MultiplyAddDivideOffsetCalculation(per_shard_window_counts[i], 0, in ReshardAsWindowedInput()
793 MultiplyAddDivideOffsetCalculation shard_limit_of_previous_on_padded( in ReshardAsWindowedInput()
799 MultiplyAddDivideOffsetCalculation shard_start_of_next_on_padded( in ReshardAsWindowedInput()
1542 MultiplyAddDivideOffsetCalculation( in HandleConcatenate()
1569 start_indices[dimension] = MultiplyAddDivideOffsetCalculation( in HandleConcatenate()
[all …]
|