TensorRT 8.5.3
NvUtils.h
Go to the documentation of this file.
1/*
2 * SPDX-FileCopyrightText: Copyright (c) 1993-2022 NVIDIA CORPORATION & AFFILIATES. All rights reserved.
3 * SPDX-License-Identifier: LicenseRef-NvidiaProprietary
4 *
5 * NVIDIA CORPORATION, its affiliates and licensors retain all intellectual
6 * property and proprietary rights in and to this material, related
7 * documentation and any modifications thereto. Any use, reproduction,
8 * disclosure or distribution of this material and related documentation
9 * without an express license agreement from NVIDIA CORPORATION or
10 * its affiliates is strictly prohibited.
11 */
12
13#ifndef NV_UTILS_H
14#define NV_UTILS_H
15
16#include "NvInfer.h"
17
23
24namespace nvinfer1
25{
26namespace utils
27{
28
76 Weights const& input, int32_t const* shape, int32_t const* shapeOrder, void* data, int32_t nbDims) noexcept;
77
124 void* input, int32_t const* order, int32_t num, int32_t size) noexcept;
125
142 void* input, DataType type, int32_t num, int32_t height, int32_t width) noexcept;
143
144} // namespace utils
145} // namespace nvinfer1
146#endif // NV_UTILS_H
#define TENSORRTAPI
Definition: NvInferRuntimeCommon.h:54
#define TRT_DEPRECATED
Definition: NvInferRuntimeCommon.h:40
An array of weights used as a layer parameter.
Definition: NvInferRuntime.h:126
TRT_DEPRECATED bool reshapeWeights(Weights const &input, int32_t const *shape, int32_t const *shapeOrder, void *data, int32_t nbDims) noexcept
Reformat the input weights of the given shape based on the new order of dimensions.
TRT_DEPRECATED bool transposeSubBuffers(void *input, DataType type, int32_t num, int32_t height, int32_t width) noexcept
Transpose num sub-buffers of height * width.
TRT_DEPRECATED bool reorderSubBuffers(void *input, int32_t const *order, int32_t num, int32_t size) noexcept
Takes an input stream and re-orders num chunks of the data given the size and order.
The TensorRT API version 1 namespace.
DataType
The type of weights and tensors.
Definition: NvInferRuntimeCommon.h:117

  Copyright © 2024 NVIDIA Corporation
  Privacy Policy | Manage My Privacy | Do Not Sell or Share My Data | Terms of Service | Accessibility | Corporate Policies | Product Security | Contact