Function calculate_subsampled_shape

arm_compute/core/Validate.h:406–428

Calculate subsampled shape for a given format and channel * * @param[in] shape Shape of the tensor to calculate the extracted channel. * @param[in] format Format of the tensor. * @param[in] channel Channel to create tensor shape to be extracted. * * @return The subsampled tensor shape. */

 * @return The subsampled tensor shape.
 */
inline TensorShape
calculate_subsampled_shape(const TensorShape &shape, Format format, Channel channel = Channel::UNKNOWN)
{
    TensorShape output{shape};

    // Subsample shape only for U or V channel
    if (Channel::U == channel || Channel::V == channel || Channel::UNKNOWN == channel)
    {
        // Subsample width for the tensor shape when channel is U or V
        if (has_format_horizontal_subsampling(format))
        {
            output.set(0, output.x() / 2U);
        }

        // Subsample height for the tensor shape when channel is U or V
        if (has_format_vertical_subsampling(format))
        {
            output.set(1, output.y() / 2U);
        }
    }

    return output;
}

/** Return an error if the passed tensor objects are not sub-sampled.
 *