NvInferLegacyDims.h

Go to the documentation of this file.

/*
 * SPDX-FileCopyrightText: Copyright (c) 1993-2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
 *
 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
 * property and proprietary rights in and to this material, related
 * documentation and any modifications thereto. Any use, reproduction,
 * disclosure or distribution of this material and related documentation
 * without an express license agreement from NVIDIA CORPORATION or
 * its affiliates is strictly prohibited.
 */
 
#ifndef NV_INFER_LEGACY_DIMS_H
#define NV_INFER_LEGACY_DIMS_H
 
#include "NvInferRuntimeCommon.h"
 
 
namespace nvinfer1
{
class Dims2 : public Dims
{
public:
    Dims2()
        : Dims2(0, 0)
    {
    }
 
    Dims2(int32_t d0, int32_t d1)
    {
        nbDims = 2;
        d[0] = d0;
        d[1] = d1;
        for (int32_t i{nbDims}; i < Dims::MAX_DIMS; ++i)
        {
            d[i] = 0;
        }
    }
};
 
class DimsHW : public Dims2
{
public:
    DimsHW()
        : Dims2()
    {
    }
 
    DimsHW(int32_t height, int32_t width)
        : Dims2(height, width)
    {
    }
 
    int32_t& h()
    {
        return d[0];
    }
 
    int32_t h() const
    {
        return d[0];
    }
 
    int32_t& w()
    {
        return d[1];
    }
 
    int32_t w() const
    {
        return d[1];
    }
};
 
class Dims3 : public Dims2
{
public:
    Dims3()
        : Dims3(0, 0, 0)
    {
    }
 
    Dims3(int32_t d0, int32_t d1, int32_t d2)
        : Dims2(d0, d1)
    {
        nbDims = 3;
        d[2] = d2;
    }
};
 
class Dims4 : public Dims3
{
public:
    Dims4()
        : Dims4(0, 0, 0, 0)
    {
    }
 
    Dims4(int32_t d0, int32_t d1, int32_t d2, int32_t d3)
        : Dims3(d0, d1, d2)
    {
        nbDims = 4;
        d[3] = d3;
    }
};
 
} // namespace nvinfer1
 
#endif // NV_INFER_LEGCY_DIMS_H