nvneural::TensorFormat Struct Reference

TensorFormat describes a specific tensor shape (element type and layout). More...

#include <nvneural/CoreTypes.h>

Public Member Functions
bool	operator!= (const TensorFormat &other) const noexcept
	Inequality operator for convenience.
bool	operator< (const TensorFormat &other) const noexcept
	Less-than operator for convenience. More...
bool	operator== (const TensorFormat &other) const noexcept
	Equality operator for convenience.
	TensorFormat ()
	Convenience default constructor. Sets everything to "invalid.".
	TensorFormat (TensorDataType elementType_, TensorDataLayout layout_)
	Convenience constructor.

Public Attributes
TensorDataType	elementType
	Element type (e.g., Float)
TensorDataLayout	layout
	Element layout (e.g., NCHW)

Detailed Description

TensorFormat describes a specific tensor shape (element type and layout).

Member Function Documentation

operator<()

bool nvneural::TensorFormat::operator< ( const TensorFormat &  other ) const

inlinenoexcept

Less-than operator for convenience.

The partial ordering it defines is suitable for std::map.

---

The documentation for this struct was generated from the following file:

• Core/Inc/nvneural/CoreTypes.h