Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dspmd_partitioner_util.h161 int64 MaxInRange(int64 start_ordinal, int64 limit_ordinal) const;
205 int64 MaxInRange(int64 start_ordinal, int64 limit_ordinal) const;
Dspmd_partitioner_util.cc581 right_halo_size_function.MaxInRange(0, src_shard_count - 1); in PadFromPartialReplicateShape()
711 int64 MultiplyAddDivideOffsetCalculation::MaxInRange( in MaxInRange() function in xla::spmd::MultiplyAddDivideOffsetCalculation
783 int64 OffsetCalculation::MaxInRange(int64 start_ordinal, in MaxInRange() function in xla::spmd::OffsetCalculation
809 int64 max_left_halo_size = left_halo_size_function.MaxInRange(1, shard_count); in ExchangeHalo()
811 right_halo_size_function.MaxInRange(0, shard_count - 1); in ExchangeHalo()
940 int64 max_left_halo_size = left_halo_size_function.MaxInRange(1, shard_count); in ExchangeHaloAndGetValidData()
961 start_offset_on_padded_concat_calculation.MaxInRange(0, shard_count) + in ExchangeHaloAndGetValidData()
Dconvolution_handler.cc395 left_halo_size_functions[rhs_dimension].MaxInRange(1, shard_counts[i]) + in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
396 right_halo_size_functions[rhs_dimension].MaxInRange( in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
420 new_padding_low_function.MaxInRange(0, shard_counts[i]); in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
Dspmd_partitioner.cc637 int64 max_shard_size = shard_size_function.MaxInRange(0, shard_count); in ReshardAsWindowedInput()
3280 (limit_window[i] - first_window[i]).MaxInRange(0, shard_count); in HandleSelectAndScatter()