Template Class PreprocessFILStage

Base Type

  • public mrc::pymrc::PythonNode< std::shared_ptr< InputT >, std::shared_ptr< OutputT > >

template<typename InputT, typename OutputT>
class PreprocessFILStage : public mrc::pymrc::PythonNode<std::shared_ptr<InputT>, std::shared_ptr<OutputT>>

FIL input data for inference.

Public Types

using base_t = mrc::pymrc::PythonNode<std::shared_ptr<InputT>, std::shared_ptr<OutputT>>

Public Functions

PreprocessFILStage(const std::vector<std::string> &features)

Constructor for a class PreprocessFILStage


features – : Reference to the features that are required for model inference

source_type_t on_data(sink_type_t x)

Called every time a message is passed to this stage

Previous Template Class PreallocateStage
Next Template Class PreprocessNLPStage
© Copyright 2024, NVIDIA. Last updated on Apr 25, 2024.