Template Class DeserializeStage

Base Type

  • public mrc::pymrc::PythonNode< std::shared_ptr< MessageMeta >, std::shared_ptr< OutputT > >

template<typename OutputT>
class DeserializeStage : public mrc::pymrc::PythonNode<std::shared_ptr<MessageMeta>, std::shared_ptr<OutputT>>

Public Types

using base_t = mrc::pymrc::PythonNode<std::shared_ptr<MessageMeta>, std::shared_ptr<OutputT>>

Public Functions

inline DeserializeStage(TensorIndex batch_size, bool ensure_sliceable_index = true, std::unique_ptr<cm_task_t> task = nullptr)

Construct a new Deserialize Stage object.

Parameters
  • batch_size – Number of messages to be divided into each batch

  • ensure_sliceable_index – Whether or not to call ensure_sliceable_index() on all incoming MessageMeta

  • task – Optional task to be added to all outgoing ControlMessages, ignored when OutputT is MultiMessage

Previous Template Class DerivedMultiMessage< DerivedT, BaseT >
Next Class DevMemInfo
© Copyright 2024, NVIDIA. Last updated on Apr 25, 2024.