TensorRT  7.2.1.6
NVIDIA TensorRT
Looking for a C++ dev who knows TensorRT?
I'm looking for work. Hire me!
All Classes Namespaces Functions Variables Typedefs Enumerations Enumerator Friends Pages
helpers.calibrator.BertCalibrator Class Reference
Inheritance diagram for helpers.calibrator.BertCalibrator:
Collaboration diagram for helpers.calibrator.BertCalibrator:

Public Member Functions

def __init__ (self, squad_json, vocab_file, cache_file, batch_size, max_seq_length, num_inputs)
 
def free (self)
 
def get_batch_size (self)
 
def get_batch (self, names)
 
def read_calibration_cache (self)
 
def write_calibration_cache (self, cache)
 
def get_quantile (self)
 
def get_regression_cutoff (self)
 
def read_histogram_cache (self, length)
 
def write_histogram_cache (self, ptr, length)
 

Public Attributes

 cache_file
 
 data
 
 max_seq_length
 
 batch_size
 
 current_index
 
 num_inputs
 
 tokenizer
 
 doc_stride
 
 max_query_length
 
 device_inputs
 

Constructor & Destructor Documentation

◆ __init__()

def helpers.calibrator.BertCalibrator.__init__ (   self,
  squad_json,
  vocab_file,
  cache_file,
  batch_size,
  max_seq_length,
  num_inputs 
)

Member Function Documentation

◆ free()

def helpers.calibrator.BertCalibrator.free (   self)
Here is the caller graph for this function:

◆ get_batch_size()

def helpers.calibrator.BertCalibrator.get_batch_size (   self)

◆ get_batch()

def helpers.calibrator.BertCalibrator.get_batch (   self,
  names 
)

◆ read_calibration_cache()

def helpers.calibrator.BertCalibrator.read_calibration_cache (   self)

◆ write_calibration_cache()

def helpers.calibrator.BertCalibrator.write_calibration_cache (   self,
  cache 
)

◆ get_quantile()

def helpers.calibrator.BertCalibrator.get_quantile (   self)

◆ get_regression_cutoff()

def helpers.calibrator.BertCalibrator.get_regression_cutoff (   self)

◆ read_histogram_cache()

def helpers.calibrator.BertCalibrator.read_histogram_cache (   self,
  length 
)

◆ write_histogram_cache()

def helpers.calibrator.BertCalibrator.write_histogram_cache (   self,
  ptr,
  length 
)

Member Data Documentation

◆ cache_file

helpers.calibrator.BertCalibrator.cache_file

◆ data

helpers.calibrator.BertCalibrator.data

◆ max_seq_length

helpers.calibrator.BertCalibrator.max_seq_length

◆ batch_size

helpers.calibrator.BertCalibrator.batch_size

◆ current_index

helpers.calibrator.BertCalibrator.current_index

◆ num_inputs

helpers.calibrator.BertCalibrator.num_inputs

◆ tokenizer

helpers.calibrator.BertCalibrator.tokenizer

◆ doc_stride

helpers.calibrator.BertCalibrator.doc_stride

◆ max_query_length

helpers.calibrator.BertCalibrator.max_query_length

◆ device_inputs

helpers.calibrator.BertCalibrator.device_inputs

The documentation for this class was generated from the following file: