holoscan::utils::set_activation_per_model

Beta
View as Markdown
gxf_result_t holoscan::utils::set_activation_per_model(
std::shared_ptr<HoloInfer::InferenceSpecs> &inference_specs,
const HoloInfer::Mappings &activation_map,
const std::vector<HoloInfer::ActivationSpec> &activation_specs,
const std::string &module
)

Setting up activation for each model by activation_map and ActivationSpec.

Parameters

inference_specs
std::shared_ptr<HoloInfer::InferenceSpecs> &

Inference spec that are needed to set the it’s activation_map.

activation_map
const HoloInfer::Mappings &

The predefined activation_map comes from the InferenceOp’s activation_map parameter.

activation_specs
const std::vector<HoloInfer::ActivationSpec> &

Activation specs received from the input port model_activation_specs of InferenceOp.

module
const std::string &

Module that called for setting up activation specifications