MCPcopy Create free account
hub / github.com/LitLeo/TensorRT_Tutorial / bert_model

Function bert_model

shenlan_homework_bert/builder.py:190–199  ·  view source on GitHub ↗

Create the bert model

(network_helper, config, weights_dict, input_tensor, input_mask)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

build_engineFunction · 0.85

Calls 1

transformer_layerFunction · 0.85

Tested by

no test coverage detected