Skip to content

Struct FNNEBufferAllocator

Struct FNNEBufferAllocator

  • Defined in File NNEBufferAllocator.h

Inheritance Relationships

Base Type

struct FNNEBufferAllocator : public ConstSpaceVisitor

Source: Source/Schola/NNE/Public/NNEUtils/NNEBufferAllocator.h

Dependencies: FBoxSpace, FDictSpace, FDiscreteSpace, FMultiBinarySpace, FMultiDiscreteSpace, FNNEPointBuffer, FSpace

Visitor that allocates appropriate buffer types based on space definitions.

This visitor traverses space definitions and creates corresponding buffer instances with appropriate sizes and types for neural network inference.


Public Functions

SymbolDetails
FNNEBufferAllocatorConstructor that takes a reference to the output buffer.
operator()Visit operation for dictionary spaces.
operator()Visit operation for box spaces.
operator()Visit operation for multi-binary spaces.
operator()Visit operation for discrete spaces.
operator()Visit operation for multi-discrete spaces.

FNNEBufferAllocator

inline FNNEBufferAllocator(TInstancedStruct<FNNEPointBuffer> &OutBuffer)

Constructor that takes a reference to the output buffer.

Parameters

OutBuffer – [out] The buffer to be allocated based on space definition

#DirectionNameTypeDescription
1OutBufferTInstancedStruct< FNNEPointBuffer > &The buffer to be allocated based on space definition

Attributes: inline

Source: Source/Schola/NNE/Public/NNEUtils/NNEBufferAllocator.h


operator()

virtual void operator()(const FDictSpace &InSpace) override

Visit operation for dictionary spaces.

Parameters

InSpace – [in] The dictionary space definition

#DirectionNameTypeDescription
1InSpaceconst FDictSpace &The dictionary space definition

Attributes: virtual

Source: Source/Schola/NNE/Public/NNEUtils/NNEBufferAllocator.h

Implementation: Source/Schola/NNE/Private/NNEUtils/NNEBufferAllocator.cpp


operator()

virtual void operator()(const FBoxSpace &InSpace) override

Visit operation for box spaces.

Parameters

InSpace – [in] The box space definition

#DirectionNameTypeDescription
1InSpaceconst FDictSpace &The dictionary space definition

Attributes: virtual

Source: Source/Schola/NNE/Public/NNEUtils/NNEBufferAllocator.h

Implementation: Source/Schola/NNE/Private/NNEUtils/NNEBufferAllocator.cpp


operator()

virtual void operator()(const FMultiBinarySpace &InSpace) override

Visit operation for multi-binary spaces.

Parameters

InSpace – [in] The multi-binary space definition

#DirectionNameTypeDescription
1InSpaceconst FDictSpace &The dictionary space definition

Attributes: virtual

Source: Source/Schola/NNE/Public/NNEUtils/NNEBufferAllocator.h

Implementation: Source/Schola/NNE/Private/NNEUtils/NNEBufferAllocator.cpp


operator()

virtual void operator()(const FDiscreteSpace &InSpace) override

Visit operation for discrete spaces.

Parameters

InSpace – [in] The discrete space definition

#DirectionNameTypeDescription
1InSpaceconst FDictSpace &The dictionary space definition

Attributes: virtual

Source: Source/Schola/NNE/Public/NNEUtils/NNEBufferAllocator.h

Implementation: Source/Schola/NNE/Private/NNEUtils/NNEBufferAllocator.cpp


operator()

virtual void operator()(const FMultiDiscreteSpace &InSpace) override

Visit operation for multi-discrete spaces.

Parameters

InSpace – [in] The multi-discrete space definition

#DirectionNameTypeDescription
1InSpaceconst FDictSpace &The dictionary space definition

Attributes: virtual

Source: Source/Schola/NNE/Public/NNEUtils/NNEBufferAllocator.h

Implementation: Source/Schola/NNE/Private/NNEUtils/NNEBufferAllocator.cpp


Public Members

SymbolDetails
BufferReference to the buffer being allocated.

Buffer

TInstancedStruct<FNNEPointBuffer> &Buffer

Reference to the buffer being allocated.


Public Static Functions

SymbolDetails
AllocateBufferStatic utility function to allocate a buffer based on a space definition.

AllocateBuffer

static inline void AllocateBuffer(const TInstancedStruct<FSpace> &InSpace,
TInstancedStruct<FNNEPointBuffer> &OutBuffer)

Static utility function to allocate a buffer based on a space definition.

Parameters

  • InSpace – [in] The space definition to allocate a buffer for

  • OutBuffer – [out] The allocated buffer

#DirectionNameTypeDescription
1InSpaceconst TInstancedStruct< FSpace > &The space definition to allocate a buffer for
2OutBufferTInstancedStruct< FNNEPointBuffer > &The allocated buffer

Attributes: inline, static

Source: Source/Schola/NNE/Public/NNEUtils/NNEBufferAllocator.h