NvInferLegacyDims.h

前往此文件的文档。

/*
 * SPDX-FileCopyrightText: Copyright (c) 1993-2024 NVIDIA CORPORATION & AFFILIATES。保留所有权利。
 * SPDX-License-Identifier: Apache-2.0
 *
 * 根据 Apache 许可证 2.0 版本（“许可证”）获得许可；
 * 除非遵守许可证，否则您不得使用此文件。
 * 您可以在以下位置获得许可证副本：
 *
 * https://apache.ac.cn/licenses/LICENSE-2.0
 *
 * 除非适用法律要求或书面同意，否则
 * 根据许可证分发的软件是按“现状”基础分发的，
 * 不附带任何明示或暗示的保证或条件。
 * 有关管理权限和
 * 许可证限制的具体语言，请参阅许可证。
 */
 
#ifndef NV_INFER_LEGACY_DIMS_H
#define NV_INFER_LEGACY_DIMS_H
 
#define NV_INFER_INTERNAL_INCLUDE 1
#include "NvInferRuntimeBase.h"
#undef NV_INFER_INTERNAL_INCLUDE
 
 
namespace nvinfer1
{
class Dims2 : public Dims
{
public
 Dims2()
 : Dims2(0, 0)
    {
    }
 
 Dims2(int64_t d0, int64_t d1)
    {
 nbDims = 2;
 d[0] = d0;
 d[1] = d1;
 for (int64_t i{nbDims}; i < Dims::MAX_DIMS; ++i)
        {
 d[i] = 0;
        }
    }
};
 
class DimsHW : public Dims2
{
public
 DimsHW()
 : Dims2()
    {
    }
 
 DimsHW(int64_t height, int64_t width)
 : Dims2(height, width)
    {
    }
 
 int64_t& h()
    {
 return d[0];
    }
 
 int64_t h() const
    {
 return d[0];
    }
 
 int64_t& w()
    {
 return d[1];
    }
 
 int64_t w() const
    {
 return d[1];
    }
};
 
class Dims3 : public Dims2
{
public
 Dims3()
 : Dims3(0, 0, 0)
    {
    }
 
 Dims3(int64_t d0, int64_t d1, int64_t d2)
 : Dims2(d0, d1)
    {
 nbDims = 3;
 d[2] = d2;
    }
};
 
class Dims4 : public Dims3
{
public
 Dims4()
 : Dims4(0, 0, 0, 0)
    {
    }
 
 Dims4(int64_t d0, int64_t d1, int64_t d2, int64_t d3)
 : Dims3(d0, d1, d2)
    {
 nbDims = 4;
 d[3] = d3;
    }
};
 
} // namespace nvinfer1
 
#endif // NV_INFER_LEGCY_DIMS_H