hub / github.com/LitLeo/TensorRT_Tutorial / addScale

Method addScale

shenlan_homework_bert/trt_helper.py:166–176 · view source on GitHub ↗

scale

(
            self,
            x: trt.ITensor,
            scale: float,
            layer_name: str = None,
            precision: trt.DataType = None
    )

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

self_attention_layerFunction · 0.80

Calls

no outgoing calls

Tested by

no test coverage detected