Home
last modified time | relevance | path

Searched refs:explicit_left_padding_on_full_shape (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/spmd/
Dconvolution_handler.cc466 int64 explicit_left_padding_on_full_shape = in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS() local
480 rhs_shard_sizes[i], explicit_left_padding_on_full_shape, 1)) - in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
487 right_halo_size_functions[dim], explicit_left_padding_on_full_shape, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnRHS()
693 int64 explicit_left_padding_on_full_shape = padding_low; in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS() local
714 right_halo_size_functions[dim], explicit_left_padding_on_full_shape, in PartitionConvolutionWithSpatialDimensionHaloExchangeOnLHS()
Dspmd_partitioner_util.cc926 int64 explicit_left_padding_on_full_shape, int64 padded_full_shape_size, in ExchangeHaloAndGetValidData() argument
947 explicit_left_padding_on_full_shape); in ExchangeHaloAndGetValidData()
995 explicit_left_padding_on_full_shape) { in ExchangeHaloAndGetValidData()
1018 explicit_left_padding_on_full_shape; in ExchangeHaloAndGetValidData()
1020 if (explicit_left_padding_on_full_shape > 0 || total_right_padding > 0) { in ExchangeHaloAndGetValidData()
1031 if (explicit_left_padding_on_full_shape > 0) { in ExchangeHaloAndGetValidData()
1037 explicit_left_padding_on_full_shape))), in ExchangeHaloAndGetValidData()
1050 explicit_left_padding_on_full_shape))), in ExchangeHaloAndGetValidData()
Dspmd_partitioner_util.h256 int64 explicit_left_padding_on_full_shape, int64 padded_full_shape_size,