hub / github.com/LitLeo/TensorRT_Tutorial / addShuffle

Method addShuffle

shenlan_homework_bert/trt_helper.py:195–222 · view source on GitHub ↗

(
        self,
        x: trt.ITensor,
        first_transpose: trt.Permutation,
        reshape_dims: trt.Dims,
        second_transpose: trt.Permutation,
        layer_name: Optional[str] = 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 1

layer_post_processMethod · 0.95

Tested by

no test coverage detected