Apollo
6.0
Open source self driving car software
|
This is the complete list of members for apollo::perception::inference::RTNet, including all inherited members.
addActiveLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addArgmaxLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addBatchnormLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, WeightMap *weight_map, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addConcatLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addConvLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, WeightMap *weight_map, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addDeconvLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, WeightMap *weight_map, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addDFMBPSROIAlignLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addEltwiseLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addInnerproductLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, WeightMap *weight_map, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addInput(const TensorDimsMap &tensor_dims_map, const std::map< std::string, std::vector< int >> &shapes, TensorMap *tensor_map) | apollo::perception::inference::RTNet | protected |
addLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, WeightMap *weight_map, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addPaddingLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addPermuteLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addPoolingLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addRCNNProposalLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addReshapeLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addRPNProposalSSDLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addScaleLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, WeightMap *weight_map, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addSliceLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
addSoftmaxLayer(const LayerParameter &layer_param, nvinfer1::ITensor *const *inputs, int nbInputs, nvinfer1::INetworkDefinition *net, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
checkInt8(const std::string &gpu_name, nvinfer1::IInt8Calibrator *calibrator) | apollo::perception::inference::RTNet | protected |
ConstructMap(const LayerParameter &layer_param, nvinfer1::ILayer *layer, TensorMap *tensor_map, TensorModifyMap *tensor_modify_map) | apollo::perception::inference::RTNet | protected |
get_blob(const std::string &name) override | apollo::perception::inference::RTNet | virtual |
gpu_id_ | apollo::perception::inference::Inference | protected |
Infer() override | apollo::perception::inference::RTNet | virtual |
Inference()=default | apollo::perception::inference::Inference | |
Init(const std::map< std::string, std::vector< int >> &shapes) override | apollo::perception::inference::RTNet | virtual |
init_blob(std::vector< std::string > *names) | apollo::perception::inference::RTNet | protected |
loadLayerWeights(const float *data, int size) | apollo::perception::inference::RTNet | protected |
loadLayerWeights(float data, int size) | apollo::perception::inference::RTNet | protected |
loadWeights(const std::string &model_file, WeightMap *weight_map) | apollo::perception::inference::RTNet | protected |
max_batch_size_ | apollo::perception::inference::Inference | protected |
mergeBN(int index, LayerParameter *layer_param) | apollo::perception::inference::RTNet | protected |
parse_with_api(const std::map< std::string, std::vector< int >> &shapes) | apollo::perception::inference::RTNet | protected |
RTNet(const std::string &net_file, const std::string &model_file, const std::vector< std::string > &outputs, const std::vector< std::string > &inputs) | apollo::perception::inference::RTNet | |
RTNet(const std::string &net_file, const std::string &model_file, const std::vector< std::string > &outputs, const std::vector< std::string > &inputs, nvinfer1::Int8EntropyCalibrator *calibrator) | apollo::perception::inference::RTNet | |
RTNet(const std::string &net_file, const std::string &model_file, const std::vector< std::string > &outputs, const std::vector< std::string > &inputs, const std::string &model_root) | apollo::perception::inference::RTNet | |
set_gpu_id(const int &gpu_id) | apollo::perception::inference::Inference | |
set_max_batch_size(const int &batch_size) | apollo::perception::inference::Inference | |
shape(const std::string &name, std::vector< int > *res) | apollo::perception::inference::RTNet | protected |
~Inference()=default | apollo::perception::inference::Inference | virtual |
~RTNet() | apollo::perception::inference::RTNet | virtual |