Searched refs:padding_config_dim (Results 1 – 5 of 5) sorted by relevance
181 auto padding_config_dim = padding_config.add_dimensions();182 padding_config_dim->set_edge_padding_low(0);183 padding_config_dim->set_interior_padding(0);184 padding_config_dim->set_edge_padding_high(padded_shape.dimensions(i) -
931 auto padding_config_dim = padding_config.add_dimensions(); in ExchangeHaloAndGetValidData() local932 padding_config_dim->set_interior_padding(0); in ExchangeHaloAndGetValidData()933 padding_config_dim->set_edge_padding_low(0); in ExchangeHaloAndGetValidData()934 padding_config_dim->set_edge_padding_high(0); in ExchangeHaloAndGetValidData()938 padding_config_dim->set_edge_padding_low(extra_left_padding); in ExchangeHaloAndGetValidData()939 padding_config_dim->set_edge_padding_high(extra_right_padding); in ExchangeHaloAndGetValidData()
970 auto padding_config_dim = padding_config.add_dimensions(); in ReshardAsWindowedInput() local971 padding_config_dim->set_interior_padding(0); in ReshardAsWindowedInput()977 padding_config_dim->set_edge_padding_low(0); in ReshardAsWindowedInput()978 padding_config_dim->set_edge_padding_high(0); in ReshardAsWindowedInput()982 padding_config_dim->set_edge_padding_low(explicit_left_padding[i]); in ReshardAsWindowedInput()983 padding_config_dim->set_edge_padding_high(padded_shape.dimensions(i) - in ReshardAsWindowedInput()4770 auto* padding_config_dim = padding_config.mutable_dimensions(dim); in PreprocessHlos() local4773 padding_config_dim->set_edge_padding_low(low_pad); in PreprocessHlos()4774 padding_config_dim->set_edge_padding_high(high_pad); in PreprocessHlos()
657 PaddingConfig::PaddingConfigDimension padding_config_dim; in PadVectorWithZeros() local658 padding_config_dim.set_edge_padding_low(zeros_to_prepend); in PadVectorWithZeros()659 padding_config_dim.set_edge_padding_high(zeros_to_append); in PadVectorWithZeros()660 *padding_config.add_dimensions() = padding_config_dim; in PadVectorWithZeros()
1500 auto padding_config_dim = padding_config.add_dimensions(); in HandleConcatenate() local1501 padding_config_dim->set_edge_padding_high(0); in HandleConcatenate()1502 padding_config_dim->set_edge_padding_low(0); in HandleConcatenate()1503 padding_config_dim->set_interior_padding(0); in HandleConcatenate()1506 padding_config_dim->set_edge_padding_low( in HandleConcatenate()1509 padding_config_dim->set_edge_padding_high( in HandleConcatenate()5072 auto padding_config_dim = padding_config.add_dimensions(); in HandleDynamicUpdateSlice() local5093 padding_config_dim->set_edge_padding_low(*beg); in HandleDynamicUpdateSlice()5094 padding_config_dim->set_edge_padding_high(bcast_width - in HandleDynamicUpdateSlice()5097 padding_config_dim->set_interior_padding(0); in HandleDynamicUpdateSlice()