Class DeserializeStage#

Inheritance Relationships#

Base Type#

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

Class Documentation#

class DeserializeStage : public mrc::pymrc::PythonNode<std::shared_ptr<MessageMeta>, std::shared_ptr<ControlMessage>>#

Public Types

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

Public Functions

inline DeserializeStage(
TensorIndex batch_size,
bool ensure_sliceable_index = true,
std::unique_ptr<control_message_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