MCPcopy Create free account
hub / github.com/NVIDIA/TensorRT / Dims2

Class Dims2

include/NvInferLegacyDims.h:36–63  ·  view source on GitHub ↗

\class Dims2 \brief Descriptor for two-dimensional data.

Source from the content-addressed store, hash-verified

34//! \brief Descriptor for two-dimensional data.
35//!
36class Dims2 : public Dims
37{
38public:
39 //!
40 //! \brief Construct an empty Dims2 object.
41 //!
42 Dims2()
43 : Dims2(0, 0)
44 {
45 }
46
47 //!
48 //! \brief Construct a Dims2 from 2 elements.
49 //!
50 //! \param d0 The first element.
51 //! \param d1 The second element.
52 //!
53 Dims2(int32_t d0, int32_t d1)
54 {
55 nbDims = 2;
56 d[0] = d0;
57 d[1] = d1;
58 for (int32_t i{nbDims}; i < Dims::MAX_DIMS; ++i)
59 {
60 d[i] = 0;
61 }
62 }
63};
64
65//!
66//! \class DimsHW

Callers 4

addLSTMLayersMethod · 0.85
constructNetworkMethod · 0.85
prepareMethod · 0.85

Calls

no outgoing calls

Tested by

no test coverage detected