Internal change

PiperOrigin-RevId: 389672355

official/vision/beta/modeling/decoders/aspp.py

@@ -173,6 +173,7 @@ def build_aspp_decoder(
       level=decoder_cfg.level,
       dilation_rates=decoder_cfg.dilation_rates,
       num_filters=decoder_cfg.num_filters,
+      use_depthwise_convolution=decoder_cfg.use_depthwise_convolution,
       pool_kernel_size=decoder_cfg.pool_kernel_size,
       dropout_rate=decoder_cfg.dropout_rate,
       use_sync_bn=norm_activation_config.use_sync_bn,

official/vision/beta/modeling/factory.py

@@ -333,6 +333,7 @@ def build_segmentation_model(
       num_convs=head_config.num_convs,
       prediction_kernel_size=head_config.prediction_kernel_size,
       num_filters=head_config.num_filters,
+      use_depthwise_convolution=head_config.use_depthwise_convolution,
       upsample_factor=head_config.upsample_factor,
       feature_fusion=head_config.feature_fusion,
       low_level=head_config.low_level,