FAST  3.2.0
Framework for Heterogeneous Medical Image Computing and Visualization
fast::TensorRTEngine Member List

This is the complete list of members for fast::TensorRTEngine, including all inherited members.

addInputNode(uint portID, std::string name, NodeType type=NodeType::IMAGE, TensorShape shape={})fast::InferenceEnginevirtual
addOutputNode(uint portID, std::string name, NodeType type=NodeType::IMAGE, TensorShape shape={})fast::InferenceEnginevirtual
getDeviceList()fast::InferenceEnginevirtual
getFilename() constfast::InferenceEnginevirtual
getInputNode(std::string name) constfast::InferenceEnginevirtual
getInputNodes() constfast::InferenceEnginevirtual
getMaxBatchSize() constfast::TensorRTEngine
fast::InferenceEngine::getMaxBatchSize()fast::InferenceEnginevirtual
getName() const overridefast::TensorRTEnginevirtual
getOutputData(std::string inputNodeName)fast::InferenceEnginevirtual
getOutputNode(std::string name) constfast::InferenceEnginevirtual
getOutputNodes() constfast::InferenceEnginevirtual
getPreferredImageOrdering() const overridefast::TensorRTEnginevirtual
getPreferredModelFormat() constfast::TensorRTEngineinlinevirtual
getReporter()fast::Object
getStaticNameOfClass()fast::Objectinlinestatic
getSupportedModelFormats() constfast::TensorRTEngineinlinevirtual
isLoaded() constfast::InferenceEnginevirtual
isModelFormatSupported(ModelFormat format)fast::InferenceEnginevirtual
load() overridefast::TensorRTEnginevirtual
m_deviceIndexfast::InferenceEngineprotected
m_deviceTypefast::InferenceEngineprotected
m_maxBatchSizefast::InferenceEngineprotected
m_modelfast::InferenceEngineprotected
m_weightsfast::InferenceEngineprotected
mInputNodesfast::InferenceEngineprotected
mOutputNodesfast::InferenceEngineprotected
mPtrfast::Objectprotected
Object()fast::Object
pointer typedeffast::InferenceEngine
reportEnd() constfast::Objectprotected
reportError()fast::Objectprotected
reportInfo()fast::Objectprotected
reportWarning()fast::Objectprotected
run() overridefast::TensorRTEnginevirtual
setDevice(int index=-1, InferenceDeviceType type=InferenceDeviceType::ANY)fast::InferenceEnginevirtual
setDeviceType(InferenceDeviceType type)fast::InferenceEnginevirtual
setFilename(std::string filename)fast::InferenceEnginevirtual
setInputData(std::string inputNodeName, std::shared_ptr< Tensor > tensor)fast::InferenceEnginevirtual
setInputNodeShape(std::string name, TensorShape shape)fast::InferenceEnginevirtual
setIsLoaded(bool loaded)fast::InferenceEngineprotectedvirtual
setMaxBatchSize(int maxBathSize)fast::TensorRTEnginevirtual
setModelAndWeights(std::vector< uint8_t > model, std::vector< uint8_t > weights)fast::InferenceEnginevirtual
setOutputNodeShape(std::string name, TensorShape shape)fast::InferenceEnginevirtual
TensorRTEngine()fast::TensorRTEngine
~Object()fast::Objectinlinevirtual
~TensorRTEngine() overridefast::TensorRTEngine