Class LSTMTensorRTInferenceOp
Defined in File lstm_tensor_rt_inference.hpp
Base Type
public holoscan::ops::GXFOperator
(Class GXFOperator)
-
class LSTMTensorRTInferenceOp : public holoscan::ops::GXFOperator
Operator class to perform the inference of the LSTM model.
This wraps a GXF Codelet(
nvidia::holoscan::custom_lstm_inference::TensorRtInference
).Public Functions
-
template<typename ArgT, typename ...ArgsT, typename = std::enable_if_t<!std::is_base_of_v<Operator, std::decay_t<ArgT>> && (std::is_same_v<Arg, std::decay_t<ArgT>> || std::is_same_v<ArgList, std::decay_t<ArgT>> || std::is_base_of_v<holoscan::Condition, typename holoscan::type_info<ArgT>::element_type> || std::is_base_of_v<holoscan::Resource, typename holoscan::type_info<ArgT>::element_type>)>>
inline LSTMTensorRTInferenceOp(ArgT &&arg, ArgsT&&... args)
- LSTMTensorRTInferenceOp() = default
- inline virtual const char *gxf_typename() const override
-
virtual void setup(OperatorSpec &spec) override
Define the operator specification.
- Parameters:
spec – The reference to the operator specification.
-
template<typename ArgT, typename ...ArgsT, typename = std::enable_if_t<!std::is_base_of_v<Operator, std::decay_t<ArgT>> && (std::is_same_v<Arg, std::decay_t<ArgT>> || std::is_same_v<ArgList, std::decay_t<ArgT>> || std::is_base_of_v<holoscan::Condition, typename holoscan::type_info<ArgT>::element_type> || std::is_base_of_v<holoscan::Resource, typename holoscan::type_info<ArgT>::element_type>)>>