Method addSoftmax

shenlan_homework_bert/trt_helper.py:143–145 · view source on GitHub ↗

(self, x: trt.ITensor, dim: int = -1, layer_name=None, precision=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

self_attention_layerFunction · 0.80

no outgoing calls

no test coverage detected