Struct DeserializeStageInterfaceProxy#

Struct Documentation#

struct DeserializeStageInterfaceProxy#

Interface proxy, used to insulate python bindings.

Public Static Functions

static std::shared_ptr<mrc::segment::Object<DeserializeStage>> init(
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>>