Searched refs:padding_dim (Results 1 – 3 of 3) sorted by relevance
755 PaddingConfig::PaddingConfigDimension padding_dim; in RewriteInputWithDynamicPadding() local756 *padding_configs.add_dimensions() = padding_dim; in RewriteInputWithDynamicPadding()767 auto* padding_dim = padding_configs.mutable_dimensions(shape_dim); in RewriteInputWithDynamicPadding() local775 padding_dim->set_edge_padding_low(dilated_window_size); in RewriteInputWithDynamicPadding()776 padding_dim->set_edge_padding_high(window_dim->padding_high() + in RewriteInputWithDynamicPadding()778 padding_dim->set_interior_padding(window_dim->base_dilation() - 1); in RewriteInputWithDynamicPadding()783 LiteralUtil::CreateR0<int32>(padding_dim->edge_padding_low()))), in RewriteInputWithDynamicPadding()1124 PaddingConfig::PaddingConfigDimension padding_dim; in RewriteDynamicSelectAndScatterSamePadding() local1129 padding_dim.set_edge_padding_high(dilated_window_size); in RewriteDynamicSelectAndScatterSamePadding()1132 *padding_configs.add_dimensions() = padding_dim; in RewriteDynamicSelectAndScatterSamePadding()
4731 std::vector<int64> padding_dim; in ParsePaddingConfig() local4732 if (!SplitToInt64s(padding_dim_str, '_', &padding_dim) || in ParsePaddingConfig()4733 (padding_dim.size() != 2 && padding_dim.size() != 3)) { in ParsePaddingConfig()4739 dim->set_edge_padding_low(padding_dim[0]); in ParsePaddingConfig()4740 dim->set_edge_padding_high(padding_dim[1]); in ParsePaddingConfig()4741 dim->set_interior_padding(padding_dim.size() == 3 ? padding_dim[2] : 0); in ParsePaddingConfig()
800 int64 padding_dim = slice_sharding_dim; in PartitionBaseCase() local801 padded_shape.set_dimensions(padding_dim, 2); in PartitionBaseCase()806 ccw_pad_config.mutable_dimensions(padding_dim) in PartitionBaseCase()808 ccw_pad_config.mutable_dimensions(padding_dim) in PartitionBaseCase()817 cw_pad_config.mutable_dimensions(padding_dim) in PartitionBaseCase()819 cw_pad_config.mutable_dimensions(padding_dim) in PartitionBaseCase()