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 incomingMessageMeta
task – Optional task to be added to all outgoing
ControlMessage
s
-
using base_t = mrc::pymrc::PythonNode<std::shared_ptr<MessageMeta>, std::shared_ptr<ControlMessage>>