Function holoscan::utils::get_data_per_model(InputContext&, const std::vector&, HoloInfer::DataMap&, std::map>&, bool, const std::string&, cudaStream_t&)
Defined in File holoinfer_utils.hpp
-
gxf_result_t holoscan::utils::get_data_per_model(InputContext &op_input, const std::vector<std::string> &in_tensors, HoloInfer::DataMap &data_per_input_tensor, std::map<std::string, std::vector<int>> &dims_per_tensor, bool cuda_buffer_out, const std::string &module, cudaStream_t &cuda_stream_out)
Extracts data from GXF Receivers.
- Parameters
op_input – Input context
in_tensors – Input tensor names
data_per_input_tensor – Map is updated with output tensor name as key mapped to data buffer
dims_per_tensor – Map is updated with tensor name as key mapped to dimension of input tensor
cuda_buffer_out – Flag defining the location of output memory (Device or Host)
module – Module that called for data extraction
cuda_stream_out – Any stream used from the input port will be stored here.
- Returns
GXF result code