def METHOD_NAME(self, inputs, training=None):
    """Calls the Atrous Spatial Pyramid Pooling layer on an input.
    Args:
      inputs: A tensor of shape [batch, height, width, channels]
    Returns:
      A tensor of shape [batch, height, width, num_channels]
    """
    result = []
    for channel in self.aspp_parallel_channels:
        temp = ops.cast(channel(inputs, training=training), inputs.dtype)
        result.append(temp)
    result[-1] = keras.layers.Resizing(
        inputs.shape[1], inputs.shape[2], interpolation="bilinear"
    )(result[-1])
    result = ops.concatenate(result, axis=-1)
    result = self.projection(result, training=training)
    return result