This is the complete list of members for nvinfer1::plugin::FlattenConcat, including all inherited members.
attachToContext(cudnnContext *cudnnContext, cublasContext *cublasContext, IGpuAllocator *gpuAllocator) override | nvinfer1::plugin::FlattenConcat | virtual |
canBroadcastInputAcrossBatch(int inputIndex) const override | nvinfer1::plugin::FlattenConcat | |
nvinfer1::IPluginV2Ext::canBroadcastInputAcrossBatch(int32_t inputIndex) const =0 | nvinfer1::IPluginV2Ext | pure virtual |
clone() const override | nvinfer1::plugin::FlattenConcat | virtual |
configurePlugin(const Dims *inputDims, int nbInputs, const Dims *outputDims, int nbOutputs, const DataType *inputTypes, const DataType *outputTypes, const bool *inputIsBroadcast, const bool *outputIsBroadcast, PluginFormat floatFormat, int maxBatchSize) override | nvinfer1::plugin::FlattenConcat | |
nvinfer1::IPluginV2Ext::configurePlugin(const Dims *inputDims, int32_t nbInputs, const Dims *outputDims, int32_t nbOutputs, const DataType *inputTypes, const DataType *outputTypes, const bool *inputIsBroadcast, const bool *outputIsBroadcast, PluginFormat floatFormat, int32_t maxBatchSize)=0 | nvinfer1::IPluginV2Ext | pure virtual |
configureWithFormat(const Dims *, int32_t, const Dims *, int32_t, DataType, PluginFormat, int32_t) | nvinfer1::IPluginV2Ext | inlineprotectedvirtual |
copyToDevice(const void *hostData, size_t count) | nvinfer1::plugin::FlattenConcat | private |
deserializeToDevice(const char *&hostBuffer, size_t count) | nvinfer1::plugin::FlattenConcat | private |
destroy() override | nvinfer1::plugin::FlattenConcat | virtual |
detachFromContext() override | nvinfer1::plugin::FlattenConcat | virtual |
enqueue(int batchSize, const void *const *inputs, void **outputs, void *workspace, cudaStream_t stream) override | nvinfer1::plugin::FlattenConcat | |
nvinfer1::IPluginV2Ext::enqueue(int32_t batchSize, const void *const *inputs, void **outputs, void *workspace, cudaStream_t stream)=0 | nvinfer1::IPluginV2 | pure virtual |
FlattenConcat(int concatAxis, bool ignoreBatch) | nvinfer1::plugin::FlattenConcat | |
FlattenConcat(int concatAxis, bool ignoreBatch, int numInputs, int outputConcatAxis, const int *inputConcatAxis, const size_t *copySize) | nvinfer1::plugin::FlattenConcat | |
FlattenConcat(const void *data, size_t length) | nvinfer1::plugin::FlattenConcat | |
FlattenConcat()=delete | nvinfer1::plugin::FlattenConcat | |
getNbOutputs() const override | nvinfer1::plugin::FlattenConcat | virtual |
getOutputDataType(int index, const nvinfer1::DataType *inputTypes, int nbInputs) const override | nvinfer1::plugin::FlattenConcat | |
nvinfer1::IPluginV2Ext::getOutputDataType(int32_t index, const nvinfer1::DataType *inputTypes, int32_t nbInputs) const =0 | nvinfer1::IPluginV2Ext | pure virtual |
getOutputDimensions(int index, const Dims *inputs, int nbInputDims) override | nvinfer1::plugin::FlattenConcat | |
nvinfer1::IPluginV2Ext::getOutputDimensions(int32_t index, const Dims *inputs, int32_t nbInputDims)=0 | nvinfer1::IPluginV2 | pure virtual |
getPluginNamespace() const override | nvinfer1::plugin::FlattenConcat | virtual |
getPluginType() const override | nvinfer1::plugin::FlattenConcat | virtual |
getPluginVersion() const override | nvinfer1::plugin::FlattenConcat | virtual |
getSerializationSize() const override | nvinfer1::plugin::FlattenConcat | virtual |
getTensorRTVersion() const | nvinfer1::IPluginV2Ext | inlineprotectedvirtual |
getWorkspaceSize(int) const override | nvinfer1::plugin::FlattenConcat | |
nvinfer1::IPluginV2Ext::getWorkspaceSize(int32_t maxBatchSize) const =0 | nvinfer1::IPluginV2 | pure virtual |
initialize() override | nvinfer1::plugin::FlattenConcat | virtual |
isOutputBroadcastAcrossBatch(int outputIndex, const bool *inputIsBroadcasted, int nbInputs) const override | nvinfer1::plugin::FlattenConcat | |
nvinfer1::IPluginV2Ext::isOutputBroadcastAcrossBatch(int32_t outputIndex, const bool *inputIsBroadcasted, int32_t nbInputs) const =0 | nvinfer1::IPluginV2Ext | pure virtual |
mCHW | nvinfer1::plugin::FlattenConcat | private |
mConcatAxisID | nvinfer1::plugin::FlattenConcat | private |
mCopySize | nvinfer1::plugin::FlattenConcat | private |
mCublas | nvinfer1::plugin::FlattenConcat | private |
mIgnoreBatch | nvinfer1::plugin::FlattenConcat | private |
mInputConcatAxis | nvinfer1::plugin::FlattenConcat | private |
mNumInputs | nvinfer1::plugin::FlattenConcat | private |
mOutputConcatAxis | nvinfer1::plugin::FlattenConcat | private |
mPluginNamespace | nvinfer1::plugin::FlattenConcat | private |
serialize(void *buffer) const override | nvinfer1::plugin::FlattenConcat | virtual |
serializeFromDevice(char *&hostBuffer, Weights deviceWeights) const | nvinfer1::plugin::FlattenConcat | private |
setPluginNamespace(const char *pluginNamespace) override | nvinfer1::plugin::FlattenConcat | virtual |
supportsFormat(DataType type, PluginFormat format) const override | nvinfer1::plugin::FlattenConcat | virtual |
terminate() override | nvinfer1::plugin::FlattenConcat | virtual |
~FlattenConcat() override | nvinfer1::plugin::FlattenConcat | |
~IPluginV2() | nvinfer1::IPluginV2 | inlineprotectedvirtual |
~IPluginV2Ext() | nvinfer1::IPluginV2Ext | inlinevirtual |