Struct AddScoresStageInterfaceProxy

struct AddScoresStageInterfaceProxy

Interface proxy, used to insulate python bindings.

Public Static Functions

static std::shared_ptr<mrc::segment::Object<AddScoresStage<MultiResponseMessage, MultiResponseMessage>>> init_multi(mrc::segment::Builder &builder, const std::string &name, std::map<std::size_t, std::string> idx2label)

Create and initialize a AddScoresStage that receives MultiResponseMessage and emits MultiResponseMessage, and return the result.

Parameters
  • builder – : Pipeline context object reference

  • name – : Name of a stage reference

  • num_class_labels – : Number of classification labels

  • idx2label – : Index to classification labels map

Returns

std::shared_ptr<mrc::segment::Object<AddScoresStage<MultiResponseMessage, MultiResponseMessage>>>

static std::shared_ptr<mrc::segment::Object<AddScoresStage<ControlMessage, ControlMessage>>> init_cm(mrc::segment::Builder &builder, const std::string &name, std::map<std::size_t, std::string> idx2label)

Create and initialize a AddScoresStage that receives ControlMessage and emits ControlMessage, and return the result.

Parameters
  • builder – : Pipeline context object reference

  • name – : Name of a stage reference

  • num_class_labels – : Number of classification labels

  • idx2label – : Index to classification labels map

Returns

std::shared_ptr<mrc::segment::Object<AddScoresStage<ControlMessage, ControlMessage>>>

Previous Struct AddClassificationStageInterfaceProxy
Next Struct ControlMessageProxy
© Copyright 2024, NVIDIA. Last updated on Apr 25, 2024.