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

Method getTensorStrides

include/NvInferRuntime.h:2747–2750  ·  view source on GitHub ↗

\brief Return the strides of the buffer for the given tensor name. The strides are in units of elements, not components or bytes. For example, for TensorFormat::kHWC8, a stride of one spans 8 scalars. Note that strides can be different for different execution contexts with dynamic shapes. If the provided name does not map to an input or output tensor, or there are dynamic dimensions that have n

Source from the content-addressed store, hash-verified

2745 //! \warning The string tensorName must be null-terminated, and be at most 4096 bytes including the terminator.
2746 //!
2747 Dims getTensorStrides(char const* tensorName) const noexcept
2748 {
2749 return mImpl->getTensorStrides(tensorName);
2750 }
2751
2752public:
2753 //!

Callers 1

getTensorInfoMethod · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected