Function internal_get_shape

src/cpu/operators/CpuWinogradConv2d.cpp:59–68

namespace
{
inline Tensor4DShape internal_get_shape(const ITensorInfo *in)
{
    const DataLayout data_layout = in->data_layout();
    const int        in_width = in->dimension(get_data_layout_dimension_index(data_layout, DataLayoutDimension::WIDTH));
    const int in_height   = in->dimension(get_data_layout_dimension_index(data_layout, DataLayoutDimension::HEIGHT));
    const int in_channels = in->dimension(get_data_layout_dimension_index(data_layout, DataLayoutDimension::CHANNEL));
    const int in_batches  = in->dimension(get_data_layout_dimension_index(data_layout, DataLayoutDimension::BATCHES));

    return Tensor4DShape{in_batches, in_height, in_width, in_channels};
}

Status validate_arguments(const ITensorInfo   *src,
                          const ITensorInfo   *weights,