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
builder.BertConfig Class Reference

Public Member Functions

def __init__ (self, bert_config_path, use_fp16, use_int8, use_strict, use_fc2_gemm, use_int8_skipln, use_int8_multihead, use_qat)
 

Public Attributes

 num_attention_heads
 
 hidden_size
 
 intermediate_size
 
 num_hidden_layers
 
 head_size
 
 use_fp16
 
 use_int8
 
 use_fc2_gemm
 
 use_strict
 
 use_int8_skipln
 
 use_int8_multihead
 
 is_calib_mode
 
 use_qat
 

Constructor & Destructor Documentation

◆ __init__()

def builder.BertConfig.__init__ (   self,
  bert_config_path,
  use_fp16,
  use_int8,
  use_strict,
  use_fc2_gemm,
  use_int8_skipln,
  use_int8_multihead,
  use_qat 
)

Member Data Documentation

◆ num_attention_heads

builder.BertConfig.num_attention_heads

◆ hidden_size

builder.BertConfig.hidden_size

◆ intermediate_size

builder.BertConfig.intermediate_size

◆ num_hidden_layers

builder.BertConfig.num_hidden_layers

◆ head_size

builder.BertConfig.head_size

◆ use_fp16

builder.BertConfig.use_fp16

◆ use_int8

builder.BertConfig.use_int8

◆ use_fc2_gemm

builder.BertConfig.use_fc2_gemm

◆ use_strict

builder.BertConfig.use_strict

◆ use_int8_skipln

builder.BertConfig.use_int8_skipln

◆ use_int8_multihead

builder.BertConfig.use_int8_multihead

◆ is_calib_mode

builder.BertConfig.is_calib_mode

◆ use_qat

builder.BertConfig.use_qat

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