NVIDIA NvNeural SDK  2022.2
GPU inference framework for NVIDIA Nsight Deep Learning Designer
nvneural::IShuffleLayer Class Referenceabstract

IShuffleLayer is a common interface implemented by space-to-depth and depth-to-space layers. More...

#include <nvneural/layers/IShuffleLayer.h>

Inheritance diagram for nvneural::IShuffleLayer:
nvneural::IRefObject

Static Public Attributes

static const IRefObject::TypeId typeID = 0x8d83e462480b362cul
 Interface TypeId for InterfaceOf purposes.
 
- Static Public Attributes inherited from nvneural::IRefObject
static const TypeId typeID = 0x14ecc3f9de638e1dul
 Interface TypeId for InterfaceOf purposes.
 

Layer parameters

enum class  ShuffleMode : std::uint32_t { Invalid = 0 , PixelDominant = 1 , ChannelDominant = 2 }
 Describes the various shuffle types. More...
 
virtual ShuffleMode shuffleMode () const noexcept=0
 Retrieves the current shuffle mode.
 
virtual NeuralResult setShuffleMode (ShuffleMode shuffleMode) noexcept=0
 Sets a shuffle mode.
 

Additional Inherited Members

- Public Types inherited from nvneural::IRefObject
using RefCount = std::uint32_t
 Typedef used to track the number of active references to an object.
 
using TypeId = std::uint64_t
 Every interface must define a unique TypeId. This should be randomized.
 
- Public Member Functions inherited from nvneural::IRefObject
virtual RefCount addRef () const noexcept=0
 Increments the object's reference count. More...
 
virtual const void * queryInterface (TypeId interface) const noexcept=0
 This is an overloaded member function, provided for convenience. It differs from the above function only in what argument(s) it accepts.
 
virtual void * queryInterface (TypeId interface) noexcept=0
 Retrieves a new object interface pointer. More...
 
virtual RefCount release () const noexcept=0
 Decrements the object's reference count and destroy the object if the reference count reaches zero. More...
 
- Protected Member Functions inherited from nvneural::IRefObject
virtual ~IRefObject ()=default
 A protected destructor prevents accidental stack-allocation of IRefObjects or use with other smart pointer classes like std::unique_ptr.
 

Detailed Description

IShuffleLayer is a common interface implemented by space-to-depth and depth-to-space layers.

You can set parameters on these layers using this interface if loading from a parameter node is undesirable.

Member Enumeration Documentation

◆ ShuffleMode

enum nvneural::IShuffleLayer::ShuffleMode : std::uint32_t
strong

Describes the various shuffle types.

Enumerator
PixelDominant 

dominant="pixel" in XML models

ChannelDominant 

dominant="shuffle" in XML models


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