Class DeserializeStage

• Defined in File deserialize.hpp

Base Type

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

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