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

Method addEmbedding

shenlan_homework_bert/trt_helper.py:81–93  ·  view source on GitHub ↗
(self, indices, weight, layer_name=None, precision=None)

Source from the content-addressed store, hash-verified

source not stored for this graph (policy: none)

Callers 1

emb_layernormFunction · 0.80

Calls 1

layer_post_processMethod · 0.95

Tested by

no test coverage detected