Function build_engine

shenlan_homework_bert/builder.py:293–344 · view source on GitHub ↗

(workspace_size, config, weights_dict, vocab_file, calibrationCacheFile, calib_num)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

generate_calibration_cacheFunction · 0.85

mainFunction · 0.85

markOutputMethod · 0.95

freeMethod · 0.95

BertCalibratorClass · 0.90

TrtNetworkHelperClass · 0.85

emb_layernormFunction · 0.85

bert_modelFunction · 0.85

transformer_output_layerFunction · 0.85

logMethod · 0.45

no test coverage detected