Defined in File deserialize.hpp
-
struct DeserializeStageInterfaceProxy
Interface proxy, used to insulate python bindings.
Public Static Functions
-
static std::shared_ptr<mrc::segment::Object<DeserializeStage<MultiMessage>>> init_multi(mrc::segment::Builder &builder, const std::string &name, TensorIndex batch_size, bool ensure_sliceable_index)
Create and initialize a DeserializationStage that emits MultiMessage’s, and return the result.
- Parameters
builder – : Pipeline context object reference
name – : Name of a stage reference
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
- Returns
std::shared_ptr<mrc::segment::Object<DeserializeStage<MultiMessage>>>
-
static std::shared_ptr<mrc::segment::Object<DeserializeStage<ControlMessage>>> init_cm(mrc::segment::Builder &builder, const std::string &name, TensorIndex batch_size, bool ensure_sliceable_index, const pybind11::object &task_type, const pybind11::object &task_payload)
Create and initialize a DeserializationStage that emits ControlMessage’s, and return the result. If
task_type
is not None,task_payload
must also be not None, and vice versa.- Parameters
builder – : Pipeline context object reference
name – : Name of a stage reference
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_type – : Optional task type to be added to all outgoing messages
task_payload – : Optional json object describing the task to be added to all outgoing messages
- Returns
std::shared_ptr<mrc::segment::Object<DeserializeStage<ControlMessage>>>
-
static std::shared_ptr<mrc::segment::Object<DeserializeStage<MultiMessage>>> init_multi(mrc::segment::Builder &builder, const std::string &name, TensorIndex batch_size, bool ensure_sliceable_index)