Clara Holoscan v0.3.0
1.0

Class LSTMTensorRTInferenceOp

Base Type

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.

© Copyright 2022, NVIDIA. Last updated on Jun 28, 2023.