builder_varseqlen.BertConfig Class Reference

Public Member Functions
def	__init__ (self, bert_config_path, use_fp16, use_int8, use_qat, interleaved)
def	get_trt_dtype (self)

Public Attributes
	num_attention_heads
	hidden_size
	intermediate_size
	num_hidden_layers
	head_size
	use_fp16
	use_int8
	use_qat
	interleaved

Constructor & Destructor Documentation

__init__()

def builder_varseqlen.BertConfig.__init__	(		self,
			bert_config_path,
			use_fp16,
			use_int8,
			use_qat,
			interleaved
	)

Member Function Documentation

get_trt_dtype()

def builder_varseqlen.BertConfig.get_trt_dtype

(

self

)

Member Data Documentation

num_attention_heads

builder_varseqlen.BertConfig.num_attention_heads

hidden_size

builder_varseqlen.BertConfig.hidden_size

intermediate_size

builder_varseqlen.BertConfig.intermediate_size

num_hidden_layers

builder_varseqlen.BertConfig.num_hidden_layers

head_size

builder_varseqlen.BertConfig.head_size

use_fp16

builder_varseqlen.BertConfig.use_fp16

use_int8

builder_varseqlen.BertConfig.use_int8

use_qat

builder_varseqlen.BertConfig.use_qat

interleaved

builder_varseqlen.BertConfig.interleaved

---

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

• builder_varseqlen.py