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 incoming MessageMeta

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 incoming MessageMeta

• 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>>>