blace.ai
|
#include <public_ops.h>
Public Member Functions | |
InferenceOp (std::vector< char > model_bytes, std::vector< OpP > inputs, ml_core::InferenceArgsCollection inference_args, int return_index, std::string payload_folder) | |
InferenceOp (InferenceOp const &e) | |
Operator to run model inference.
blace::ops::InferenceOp::InferenceOp | ( | std::vector< char > | model_bytes, |
std::vector< OpP > | inputs, | ||
ml_core::InferenceArgsCollection | inference_args, | ||
int | return_index, | ||
std::string | payload_folder | ||
) |
Constructs an operator which runs model inference.
model_bytes | The bytes of the model to run, will be provided from the included model header. |
inputs | A vector of input nodes passed to the model. |
inference_args | The inference arguments used to run the inference. |
return_index | The desired result index, some models return multiple tensors. |
payload_folder | Payload folder. |
blace::ops::InferenceOp::InferenceOp | ( | InferenceOp const & | e | ) |
Default copy constructor explicitly defined for dll exporting.
e | The copy class. |