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

Function transformer_layer

shenlan_homework_bert/builder.py:138–164  ·  view source on GitHub ↗
(network_helper, prefix, config, weights_dict, input_tensor, imask)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

bert_modelFunction · 0.85

Calls 5

attention_layerFunction · 0.85
addLinearMethod · 0.80
addGELUMethod · 0.80
addAddMethod · 0.80
addLayerNormMethod · 0.80

Tested by

no test coverage detected