nvinfer1::Int8EntropyCalibrator类 参考

#include <entropy_calibrator.h>

类 nvinfer1::Int8EntropyCalibrator 继承关系图:

nvinfer1::Int8EntropyCalibrator 的协作图:

Public 成员函数
	Int8EntropyCalibrator (const apollo::perception::inference::BatchStream &stream, int first_batch, bool read_cache, std::string network)
virtual	~Int8EntropyCalibrator ()
int	getBatchSize () const noexcept override
bool	getBatch (void *bindings[], const char *names[], int nbBindings) noexcept override
const void *	readCalibrationCache (size_t &length) noexcept override
void	writeCalibrationCache (const void *cache, size_t length) noexcept override

详细描述

在文件 entropy_calibrator.h 第 29 行定义.

构造及析构函数说明

Int8EntropyCalibrator()

nvinfer1::Int8EntropyCalibrator::Int8EntropyCalibrator	(	const apollo::perception::inference::BatchStream &	stream,
		int	first_batch,
		bool	read_cache,
		std::string	network
	)

在文件 entropy_calibrator.cc 第 33 行定义.

    : stream_(stream), read_cache_(read_cache), network_(network) {
  DimsNCHW dims = stream_.getDims();
  input_count_ = stream_.getBatchSize() * dims.c() * dims.h() * dims.w();
  cudaMalloc(&device_input_, input_count_ * sizeof(float));
  stream_.reset(first_batch);
}

~Int8EntropyCalibrator()

nvinfer1::Int8EntropyCalibrator::~Int8EntropyCalibrator ( )

virtual

在文件 entropy_calibrator.cc 第 43 行定义.

                                              {
  if (device_input_) {
    (cudaFree(device_input_));
  }
}

成员函数说明

getBatch()

bool nvinfer1::Int8EntropyCalibrator::getBatch ( void *  bindings[], const char *  names[], int  nbBindings  )

overridenoexcept

在文件 entropy_calibrator.cc 第 49 行定义.

                                                              {
  if (!stream_.next()) {
    return false;
  }
 
  (cudaMemcpy(device_input_, stream_.getBatch(), input_count_ * sizeof(float),
              cudaMemcpyHostToDevice));
  bindings[0] = device_input_;
  return true;
}

getBatchSize()

int nvinfer1::Int8EntropyCalibrator::getBatchSize ( ) const

inlineoverridenoexcept

在文件 entropy_calibrator.h 第 36 行定义.

{ return stream_.getBatchSize(); }

readCalibrationCache()

const void * nvinfer1::Int8EntropyCalibrator::readCalibrationCache ( size_t &  length )

overridenoexcept

在文件 entropy_calibrator.cc 第 61 行定义.

                             {
  calibration_cache_.clear();
  std::ifstream input(
      apollo::perception::inference::locateFile(network_, "CalibrationTable"),
      std::ios::binary);
  input >> std::noskipws;
  if (read_cache_ && input.good()) {
    std::copy(std::istream_iterator<char>(input), std::istream_iterator<char>(),
              std::back_inserter(calibration_cache_));
  }
 
  length = calibration_cache_.size();
  return length ? &calibration_cache_[0] : nullptr;
}

writeCalibrationCache()

void nvinfer1::Int8EntropyCalibrator::writeCalibrationCache ( const void *  cache, size_t  length  )

overridenoexcept

在文件 entropy_calibrator.cc 第 77 行定义.

                                                                          {
  std::ofstream output(
      apollo::perception::inference::locateFile(network_, "CalibrationTable"),
      std::ios::binary);
  output.write(reinterpret_cast<const char *>(cache), length);
}

---

该类的文档由以下文件生成:

• modules/perception/common/inference/tensorrt/entropy_calibrator.h
• modules/perception/common/inference/tensorrt/entropy_calibrator.cc