rt_common.h File Reference

#include <cudnn.h>
#include <map>
#include <string>
#include <vector>
#include "NvCaffeParser.h"
#include "NvInfer.h"
#include "cyber/common/log.h"
#include "modules/perception/base/common.h"
#include "modules/perception/proto/rt.pb.h"

Include dependency graph for rt_common.h:

This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Classes
struct	apollo::perception::inference::ConvParam

Namespaces
	apollo
	apollo::perception
	apollo::perception::inference

Typedefs
typedef std::map< std::string, std::vector< nvinfer1::Weights > >	apollo::perception::inference::WeightMap
typedef std::map< std::string, nvinfer1::ITensor * >	apollo::perception::inference::TensorMap
typedef std::map< std::string, nvinfer1::DimsCHW >	apollo::perception::inference::TensorDimsMap

Functions
nvinfer1::DimsCHW	apollo::perception::inference::ReshapeDims (const nvinfer1::DimsCHW &dims, const nvinfer1::DimsCHW &inputDims)
void	apollo::perception::inference::ParseNetParam (const NetParameter &net_param, TensorDimsMap *tensor_dims_map, std::map< std::string, std::string > *tensor_modify_map, std::vector< LayerParameter > *order)
bool	apollo::perception::inference::modify_pool_param (PoolingParameter *pool_param)
bool	apollo::perception::inference::ParserConvParam (const ConvolutionParameter &conv, ConvParam *param)